Training Model issue

[yzzupgo]

Hello! Thanks for open-sourcing the work!
I am trying to train the model without the docker. However, I encounter the error shown in the figure.

The error is caused by line 151 of run.py. Specifically, the variable test_set is defined on line 150 of run.py, and the argument "test" is passed to SumDataset to initialize test_set. However, the third branch of the init function in the Dataset.py does not do anything, which causes self.data to be empty.
Furthermore, could you provide the files of the test set and validation set (i.e., valdata.pkl, valnl.pkl, testdata.pkl, testnl.pkl, testcopy.pkl).
Thank you for taking a look.

[pkuzqh]

you can copy them from the docker

[yzzupgo]

Thanks for your quick reply.