search

jianweiguo / SpecularityNet-PSD

35 stars 3 forks source link

训练过程中epoch始终为1 #3

Open aobi12138 opened 2 years ago

aobi12138 commented 2 years ago

您好，我在训练您的PSD_Dataset过程中，终端打印的Epoch始终为0。请问是否需要修改train_specularitynet.py第71行为 engine.epoch += 1？还是应该取消engine.py第68行的注释self.epoch += 1呢？