python run_image.py

[aboudjelal]

I want to test run_image.py function but not working, please give me a solution
Thanks

/graph-neural-pde/src$ python run_image.py
[pyKeOps]: Warning, cuda was detected, but driver API could not be initialized. Switching to cpu only.
Loading Data
creating in_memory_datasets
$anaconda3/envs/grand_2/lib/python3.7/site-packages/torch_geometric/deprecation.py:13: UserWarning: 'data.DataLoader' is deprecated, use 'loader.DataLoader' instead
warnings.warn(out)
creating GNN model
Traceback (most recent call last):
File "run_image.py", line 340, in
main(opt)
File "run_image.py", line 164, in main
model = GNN_image(opt, batch, opt['num_class'], device).to(device)
File "/home/graph-neural-pde/src/GNN_image.py", line 22, in init
range(self.n_ode_blocks)]).to(self.device)
File "/home/anaconda3/envs/grand_2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 948, in getattr
type(self).name, name))
AttributeError: 'GNN_image' object has no attribute 'n_ode_blocks'