NVlabs / AL-MDN

Official pytorch implementation of Active Learning for deep object detection via probabilistic modeling (ICCV 2021)
https://openaccess.thecvf.com/content/ICCV2021/html/Choi_Active_Learning_for_Deep_Object_Detection_via_Probabilistic_Modeling_ICCV_2021_paper.html
Other
166 stars 21 forks source link

Autograde issue while evalution #21

Open AlgorithmicPlanet opened 1 year ago

AlgorithmicPlanet commented 1 year ago

/home/bridgei2i/.local/lib/python3.10/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: libtorch_cudacu.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 commented 1 year ago

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"