GCC version: (Ubuntu 7.5.0-6ubuntu2) 7.5.0ĬuDNN version: Probably one of the following:

RuntimeError: expected scalar type Long but found Float Environment _jit_pass_onnx_constant_fold( graph, params_dict, export( model, args, f, export_params, verbose, training,įile '/home/space/Public/anaconda3/envs/p圓8torch19/lib/python3.8/site-packages/torch/onnx/utils.py', line 88, in export _export( model, args, f, export_params, verbose, training, input_names, output_names,įile '/home/space/Public/anaconda3/envs/p圓8torch19/lib/python3.8/site-packages/torch/onnx/utils.py', line 689, in _export _model_to_graph( model, args, verbose, input_names,įile '/home/space/Public/anaconda3/envs/p圓8torch19/lib/python3.8/site-packages/torch/onnx/utils.py', line 501, in _model_to_graph params_dict = torch. export(įile '/home/space/Public/anaconda3/envs/p圓8torch19/lib/python3.8/site-packages/torch/onnx/_init_.py', line 275, in export return utils.

File '/data2/Documents/Experiments/BaseT/models/baseline/model2trt.py', line 218, in torch.

Set model params nn torch