问题描述
我需要分别为两个不同的模型调用向后函数。
# Loss_1
#----------------------
Losses_1 = []
with autograd.record():
for batch_per_gpu in batch
es:
pred = model_1(batch_per_gpu)
loss = loss_func(pred,truth)
losses_1.append(loss)
mx.autograd.backward(losses_1)
model_1_trainer.allreduce_grads()
model_1_trainer.update()
#Loss_2
#----------------------
Losses_2 = []
with autograd.record():
for batch_per_gpu in batches:
pred = model_2(batch_per_gpu)
loss = loss_func(pred,truth)
losses_2.append(loss)
mx.autograd.backward(losses_2)
model_2_trainer.allreduce_grads()
model_2_trainer.update()
第二次调用向后函数时出现此错误:
Check failed: type_ != nullptr: The any container is empty requested=N5mxnet10Imperative6AGInfoE
有人知道如何解决此问题吗?
我的MXNET版本是1.6.0
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)