1、打印当前时间

运行程序时可能需要记录

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

• 1
• 2
• 3

打印出的结果

2、保存模型

方法1：保存网络和参数

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

• 1
• 2

读取：

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

• 1
• 2

方法2：保存网络参数

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

• 1
• 2

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

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

• 1
• 2