RuiqiZhang

cannot find the images when training my personal dataset

13

Epoch 1/300: 0%| | 0/40 [00:00

Size mismatch when loading pre-trained models

6

Hi, ### When I try to load pre-trained models to test predict.py, I was noticed as follows: python predict.py pre-trained/final-energy-per-atom.pth.tar mp/ => loading model params 'pre-trained/final-energy-per-atom.pth.tar' => loaded model params...