1、打印当前时间

运行程序时可能需要记录

    import datetime
    time = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    print(time)

打印出的结果
在这里插入图片描述

2、保存模型

方法1:保存网络和参数

torch.save(model, 'model.pth')

读取:

model = torch.load('model.pth')

方法2:保存网络参数

torch.save(model.state_dict(), 'model_params.pth')

需要实例化网络,并导入参数

model.load_state_dict(torch.load('model_params.pth'))

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐