参考:https://discuss.pytorch.org/t/runtimeerror-cannot-insert-a-tensor-that-requires-grad-as-a-constant/113767/4
我们在导出onnx模型的时候也要这样做,就可以正确导出了。看了很多资料,目前试过好用的就是这一种。
for para in model.parameters():
para.requires_grad=False
参考:https://discuss.pytorch.org/t/runtimeerror-cannot-insert-a-tensor-that-requires-grad-as-a-constant/113767/4
我们在导出onnx模型的时候也要这样做,就可以正确导出了。看了很多资料,目前试过好用的就是这一种。
for para in model.parameters():
para.requires_grad=False