maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

[PyT/ConvNets] Fixing bug in model2onnx

Browse source
This commit is contained in:
Andrzej Sulecki 2021-07-02 11:31:19 +02:00 committed by Krzysztof Kudrynski
parent 96b70f5d16
commit a1bbe6687e
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
PyTorch/Classification/ConvNets/model2onnx.py
View file

@ -113,7 +113,7 @@ def main(args, model_args, model_arch):
model = model_arch(**model_args.__dict__)
if quant_arch:
if quant_arch and model_args.pretrained_from_file is not None:
check_quant_weight_correctness(model_args.pretrained_from_file, model)
image_size = args.image_size if args.image_size is not None else model.arch.default_image_size