the item in "total_img " is nan?

[nlsde-safety-team]

Thanks for your interests! The dataset need to be processed by the function parse_npz() in the line src/nmr_test.py/282. You can refer to the function and process the dataset.