模型是DeblurGan_v2。在模型推理的时候需要保持train为true(图一)。如果我要导出为ONNX，应该怎么做？ 图二是我导出模型的代码，结果表明效果不好，和train为false的结果一样。
pytorch吧 关注：1,349贴子：2,494
- 3回复贴，共1页
设置这个参数
training (enum, default TrainingMode.EVAL) –
TrainingMode.EVAL: export the model in inference mode.
TrainingMode.PRESERVE: export the model in inference mode if model.training is
False and in training mode if model.training is True.
TrainingMode.TRAINING: export the model in training mode. Disables optimizations
which might interfere with training.
详细信息看文档，这里我随便粘贴了一下
training (enum, default TrainingMode.EVAL) –
TrainingMode.EVAL: export the model in inference mode.
TrainingMode.PRESERVE: export the model in inference mode if model.training is
False and in training mode if model.training is True.
TrainingMode.TRAINING: export the model in training mode. Disables optimizations
which might interfere with training.
详细信息看文档，这里我随便粘贴了一下
扫二维码下载贴吧客户端
下载贴吧APP
看高清直播、视频！
看高清直播、视频！
贴吧热议榜
- 1翟欣欣逼死前夫或被判10年以上1852650
- 2如何看待美国接连“退群”1317847
- 3已经给二师兄穿上JK了1009484
- 4LOL灵蛇秘宝比惨大会721143
- 5TikTok禁令推迟75天646490
- 6没想到三国无双有登顶的一天642125
- 7特朗普正式宣誓就职美国总统602592
- 8台湾台南地震多地震感明显446384
- 9外交部回应X不被允许在中国运营355344
- 10现场直击特朗普就职上任342300
