This thread has been locked.

If you have a related question, please click the "Ask a related question" button in the top right corner. The newly created question will be automatically linked to this question.

support for transpose conv



Traceback (most recent call last):
  File "main_apa_int8_train.py", line 549, in <module>
    main(parser.parse_args())
  File "main_apa_int8_train.py", line 442, in main
    model = xnn.quantize.QuantTrainModule(model,dummy_input=dummy_input)
  File "/home/data/erfnet-pytorch/train/modules/pytorch_jacinto_ai/xnn/quantize/quant_train_module.py", line 39, in __init__
    self.model_surgery_quantize(dummy_input)
  File "/home/data/erfnet-pytorch/train/modules/pytorch_jacinto_ai/xnn/quantize/quant_train_module.py", line 121, in model_surgery_quantize
    self.apply(replace_func)
  File "/usr/local/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 247, in apply
    module.apply(fn)
  File "/usr/local/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 247, in apply
    module.apply(fn)
  File "/usr/local/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 247, in apply
    module.apply(fn)
  [Previous line repeated 1 more time]
  File "/usr/local/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 248, in apply
    fn(self)
  File "/home/data/erfnet-pytorch/train/modules/pytorch_jacinto_ai/xnn/quantize/quant_train_module.py", line 87, in replace_func
    assert False, 'TODO: handle ConvTranspose2d in replace_quant_modules()'
AssertionError: TODO: handle ConvTranspose2d in replace_quant_modules()