Traceback (most recent call last):


 "test.py", line 56, in <module>

torch.onnx.export(model, dummy_input, "Deblur.onnx")


 "/Users/heojin/anaconda3/envs/tensorflow/lib/python3.7/site-packages/torch/onnx/__init__.py", line 143, in export

strip_doc_string, dynamic_axes, keep_initializers_as_inputs)


 "/Users/heojin/anaconda3/envs/tensorflow/lib/python3.7/site-packages/torch/onnx/utils.py", line 66, in export

dynamic_axes=dynamic_axes, keep_initializers_as_inputs=keep_initializers_as_inputs)


 "/Users/heojin/anaconda3/envs/tensorflow/lib/python3.7/site-packages/torch/onnx/utils.py", line 382, in _export

fixed_batch_size=fixed_batch_size)


 "/Users/heojin/anaconda3/envs/tensorflow/lib/python3.7/site-packages/torch/onnx/utils.py", line 249, in _model_to_graph

graph, torch_out = _trace_and_get_graph_from_model(model, args, training)


 "/Users/heojin/anaconda3/envs/tensorflow/lib/python3.7/site-packages/torch/onnx/utils.py", line 198, in _trace_and_get_graph_from_model

orig_state_dict_keys = _unique_state_dict(model).keys()


 "/Users/heojin/anaconda3/envs/tensorflow/lib/python3.7/site-packages/torch/jit/__init__.py", line 282, in _unique_state_dict

state_dict = module.state_dict(keep_vars=True)
AttributeError: 'TestModel' object has no attribute 'state_dict'

Traceback (most recent call last):



 "<ipython-input-186-7ad46ff9ab2f>", line 1, in <module>

runfile('/lena13/home_users/users/chaumon/owncloud/Lab/Projects/CTPS/Datalink/MNE_analysis/code/plot_forward_sensitivity_maps.py', wdir='/lena13/home_users/users/chaumon/owncloud/Lab/Projects/CTPS/Datalink/MNE_analysis/code')



 "/home/anaconda3/envs/mne/lib/python3.6/site-packages/spyder/utils/site/sitecustomize.py", line 705, in runfile

execfile(filename, namespace)



 "/home/anaconda3/envs/mne/lib/python3.6/site-packages/spyder/utils/site/sitecustomize.py", line 102, in execfile

exec(compile(f.read(), filename, 'exec'), namespace)



 "/lena13/home_users/users/chaumon/owncloud/Lab/Projects/CTPS/Datalink/MNE_analysis/code/plot_forward_sensitivity_maps.py", line 75, in <module>

report.add_figs_to_section(fig, captions=['test'], section='test')



 "/home/anaconda3/envs/mne/lib/python3.6/site-packages/mne/report.py", line 950, in add_figs_to_section

comments=comments)



 "/home/anaconda3/envs/mne/lib/python3.6/site-packages/mne/report.py", line 905, in _add_figs_to_section

img = _fig_to_img(fig, image_format, scale)



 "/home/anaconda3/envs/mne/lib/python3.6/site-packages/mne/report.py", line 77, in _fig_to_img

if fig.scene is not None:

AttributeError: 'Brain' object has no attribute 'scene'
