We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0d0f954 commit feca7bcCopy full SHA for feca7bc
advanced_source/super_resolution_with_onnxruntime.py
@@ -142,7 +142,7 @@ def _initialize_weights(self):
142
"super_resolution.onnx", # where to save the model (can be a file or file-like object)
143
export_params=True, # store the trained parameter weights inside the model file
144
opset_version=10, # the ONNX version to export the model to
145
- do_constant_folding=True, # wether to execute constant folding for optimization
+ do_constant_folding=True, # whether to execute constant folding for optimization
146
input_names = ['input'], # the model's input names
147
output_names = ['output'], # the model's output names
148
dynamic_axes={'input' : {0 : 'batch_size'}, # variable lenght axes
0 commit comments