Hello @JiaRenChang , thanks for your work. When I run the main.py, I have completed trainning for 10 epochs. But, when the test module is performed, an error is reported as follows
. And I didn't change anything in the test module. So how can I solve this problem? thank you!
Hello @JiaRenChang , thanks for your work. When I run the main.py, I have completed trainning for 10 epochs. But, when the test module is performed, an error is reported as follows . And I didn't change anything in the test module. So how can I solve this problem? thank you!