我的程序训练一段时间,进行test输出,继续训练的时候出现标题所述问题。
我遇到的问题不能通过loss.requires_grad_(True)解决。后发现原因是,我传入test的网络模型是用于train的网络模型实例。该问题的解决方案是不要将用于train的实例传入test,即假设
net = Model.net1(),则不要test不要传入net,而要传入Model.net1()
我的程序训练一段时间,进行test输出,继续训练的时候出现标题所述问题。
我遇到的问题不能通过loss.requires_grad_(True)解决。后发现原因是,我传入test的网络模型是用于train的网络模型实例。该问题的解决方案是不要将用于train的实例传入test,即假设
net = Model.net1(),则不要test不要传入net,而要传入Model.net1()