非常基础的东西,写下来以防忘记。
将损失值loss_value设置为无限大,每次迭代若损失值比loss_value小则保存模型,并将最新的损失值赋给loss_value。
loss_value = np.inf
...
if loss < loss_value:
torch.save(model, 'model.ckpt')
loss_value = loss
非常基础的东西,写下来以防忘记。
将损失值loss_value设置为无限大,每次迭代若损失值比loss_value小则保存模型,并将最新的损失值赋给loss_value。
loss_value = np.inf
...
if loss < loss_value:
torch.save(model, 'model.ckpt')
loss_value = loss