main.py - Train the model;
test_acc.py - Show training/validating/testing accuracy;
test_validate.py - Draw predicted phyiscal field to validate the results;
The dataset has been uploaded to:
Link:https://terabox.com/s/1fTr4fWd0VMD_3yUF9jfIvQ
Password:4p7i
Preprint on Arxiv:
https://arxiv.org/abs/2108.13509?context=cs.LG