如何解决连续两次向后调用时MXNET Gluon错误-检查失败:type_!= nullptr
我需要分别为两个不同的模型调用向后函数。
# 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 举报,一经查实,本站将立刻删除。