Cuteboom的博客

以上的三个部分，所有部分完成，接下来导入进训练流程中即可。model = model.to(device) # 模型和数据 ，要在一个设备上。cpu - gpumin_val_loss = 100000 # 记录训练验证loss 以及验证loss和结果for i in range(epoch): # 训练epoch 轮start_time = time.time() # 记录开始时间model.train() # 模型设置为训练状态 结构。