在代码中添加代码
writer.add_scalars('loss',{'loss':loss_all.item()},global_step=epoch)
正确应该为:writer.add_scalars(‘loss’,{‘loss’:loss_all.item()},global_step=epoch)
在代码中添加代码
writer.add_scalars('loss',{'loss':loss_all.item()},global_step=epoch)
正确应该为:writer.add_scalars(‘loss’,{‘loss’:loss_all.item()},global_step=epoch)