动态图训练时只能训练一个epoch
Created by: zehua9500
我是在AI studio上用动态图 训练模型的 for epoch in range(10): for data in train_reader(): XXXXX 这样训练时,在第一个epoch时可以看到loss逐渐下降(从1下降到0.3),但只要第一个epoch结束后,从第二个epoch开始,loss就基本一直在0.5了,不再训练(loss也变稍高)。
但若是我把第一个epoch训练完成后保存下来,重启notebook,后加载第一个epoch的参数继续训练,loss就会从0.3在慢慢下降。可下一个epoch loss又稳定在0.5左右了