pytorch 运行时错误:推理Tensor不能保存为向后,要解决此问题,您可以克隆一个正常Tensor并在autograd中使用它

dm7nw8vv  于 2023-03-02  发布在  其他
关注(0)|答案(1)|浏览(317)

时间:程序在确认后开始下一个时期
对此如何修改或设置?

bt1cpqcv

bt1cpqcv1#

也许在定义优化器、训练循环之前,您需要再次示例化模型

model_0 = TinyVGG(3, 10, len(train_data.classes)).to(device)
optimizer = torch.optim.Adam(params=model_0.parameters(), lr=0.001)

训练你的模特!

相关问题