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()