level 2
设置这个参数
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.
详细信息看文档,这里我随便粘贴了一下
2023年04月12日 10点04分

