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'))

更多推荐

python打印当前时间