search

232525 / PureT

Implementation of 'End-to-End Transformer Based Model for Image Captioning' [AAAI 2022]
63 stars 12 forks source link

保存训练恢复 #27

Open ImaginationCreater opened 1 month ago

ImaginationCreater commented 1 month ago

作者好，我在训练的过程中中断了之后，然后我看到模型保存到了9，然后我就将resume设置为9，load_epoch设置为True,然后运行程序时报错了，具体情况图如下，请问是我哪里出错了么？万分感谢。 image image image

232525 commented 1 month ago
image

应该是这，学习参数设置为注释掉的那部分 https://github.com/232525/PureT/blob/main/optimizer/optimizer.py

ImaginationCreater commented 1 month ago

好的，谢谢您。