Autograde issue while evalution

[AlgorithmicPlanet]

/home/bridgei2i/.local/lib/python3.10/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory
warn(f"Failed to load image Python extension: {e}")
/Datadisk/AILabs/ComputerVision/Shoeb/AL-MDN/layers/modules/l2norm.py:20: UserWarning: nn.init.constant is now deprecated in favor of nn.init.constant_.
init.constant(self.weight,self.gamma)
Clipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers).
/home/bridgei2i/.local/lib/python3.10/site-packages/torch/nn/functional.py:780: UserWarning: Note that order of the arguments: ceil_mode and return_indices will changeto match the args list in nn.MaxPool2d in a future release.
warnings.warn("Note that order of the arguments: ceil_mode and return_indices will change"
Traceback (most recent call last):
File "/Datadisk/AILabs/ComputerVision/Shoeb/AL-MDN/demo.py", line 70, in
y = net(xx)
File "/home/bridgei2i/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/bridgei2i/.local/lib/python3.10/site-packages/torch/nn/parallel/data_parallel.py", line 168, in forward
outputs = self.parallel_apply(replicas, inputs, kwargs)
File "/home/bridgei2i/.local/lib/python3.10/site-packages/torch/nn/parallel/data_parallel.py", line 178, in parallel_apply
return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
File "/home/bridgei2i/.local/lib/python3.10/site-packages/torch/nn/parallel/parallel_apply.py", line 86, in parallel_apply
output.reraise()
File "/home/bridgei2i/.local/lib/python3.10/site-packages/torch/_utils.py", line 457, in reraise
raise exception
RuntimeError: Caught RuntimeError in replica 0 on device 0.
Original Traceback (most recent call last):
File "/home/bridgei2i/.local/lib/python3.10/site-packages/torch/nn/parallel/parallel_apply.py", line 61, in _worker
output = module(*input, **kwargs)
File "/home/bridgei2i/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/Datadisk/AILabs/ComputerVision/Shoeb/AL-MDN/ssd_gmm.py", line 271, in forward
output = self.detect(
File "/home/bridgei2i/.local/lib/python3.10/site-packages/torch/autograd/function.py", line 315, in call
raise RuntimeError(
RuntimeError: Legacy autograd function with non-static forward method is deprecated. Please use new-style autograd function with static forward method. (Example: https://pytorch.org/docs/stable/autograd.html#torch.autograd.Function)

[AliceShynie]

This is due to your pytorch version, I used python 3.6 with torch 1.1.0. The way I used it was through conda environment. You can find the command here. This is the command I used : conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch. Okay if you do choose this route, the code uses one package, that is called pickle, you might have to change it to "import pickle5 as pickle"