我正在使用PyTorch训练我的模型,并且我已经添加了保存和恢复训练功能,但是问题是当我想覆盖张量板上的步长值时,该图会丢失(将最后一步连接到当前步骤)
示例:
我每10次迭代将标量记录到张量板上,并且每100次迭代将模型保存一次,其中迭代= step。
在下图中,我对模型进行了150次迭代训练(因此我现在在张量板图中有15个步点),然后我从另外100次迭代中恢复训练并开始在步骤100中进行记录,我希望可以覆盖步骤110之后的所有步骤点,但似乎将最后一个步骤点(150)连接到步骤110,如何覆盖步骤100之后的所有步骤,并修复图???
登录到张量板代码:
if (OLD_ITERATIONS + i)%10 is 0 and i is not 0:
wirter.add_scalar(tag='Accuracy/Train', scalar_value=accs/10, global_step=OLD_ITERATIONS+i)
wirter.add_scalar(tag='Accuracy/Valid', scalar_value=val_accs/10, global_step=OLD_ITERATIONS+i)
wirter.add_scalar(tag='Loss/Train', scalar_value=losses/10, global_step=OLD_ITERATIONS+i)
wirter.add_scalar(tag='Loss/Valid', scalar_value=val_losses/10, global_step=OLD_ITERATIONS+i)
accs = 0
losses = 0
val_accs = 0
val_losses = 0