如何解决上一个时期计算出的变量如何用于下一个时期计算损失函数?
pytorch代码如下。前一个时期计算出的变量如何用于 在下一个时期计算损失函数?
for epoch in range((args.start_epoch+1),args.epochs):
for input,target in train_loader:
target = target.cuda()
input = input.cuda()
input_var = torch.autograd.Variable(input)
target_var = torch.autograd.Variable(target)
outputs,feature = model(input_var)
if epoch > 0:
l = criterion.forward(feature,target_var,Fea)
Fea = function(model,train_loader)
解决方法
for epoch in range((args.start_epoch+1),args.epochs):
Center = computer_Center(型号,数据加载器,类编号) 作为输入,将目标设置在train_loader中: 目标= target.cuda() 输入= input.cuda() input_var = torch.autograd.Variable(输入)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。