赛事笔记4——模型保存标准格式

从朋友那里get了一个保存模型的而标准格式，防止训练过程中出现终端，发生中断从断开的epoch继续训练。

    if val_loss < best_loss:
        best_loss = val_loss
        # print('Find better model in Epoch {0}, saving model.'.format(epoch))
        # torch.save(model.state_dict(), './model.pt')
        torch.save({'epoch': epoch,
                    'model_state_dict': model.state_dict(),
                    'optimizer_state_dict': optimizer.state_dict(),
                    'loss':best_loss},
                    './output/best_model.pt')