Error.md

Command Ran:

/usr/bin/salloc -n 20 -J interactive -p ihub -A mobility_arfs --gres=gpu:2 -t 12:00:00 /usr/bin/srun  --pty --cpu_bind=no /bin/bash -l

python3 infer.py --cfg /home2/jaskirat.singh/Drone-based-building-assessment/win_det_heatmaps/experiments/shufflenet/lr1e-3_120x90-110_center_b2.yaml 
                 --model /home2/jaskirat.singh/shufflenet_zjufacade+iiitH.pth
                 --infer /home2/jaskirat.singh/002

Error:

=================SYS PATH================
/home2/jaskirat.singh/Drone-based-building-assessment/win_det_heatmaps/common_pytorch
/home2/jaskirat.singh/Drone-based-building-assessment/win_det_heatmaps/common
/home2/jaskirat.singh/Drone-based-building-assessment/win_det_heatmaps
/home2/jaskirat.singh/miniconda3/envs/robo/lib/python38.zip
/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8
/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/lib-dynload
/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages
=================SYS PATH================
Using Devices: [0]
Defining result & flip func
Creating dataset
Loading model from /home2/jaskirat.singh/shufflenet_zjufacade+iiitH.pth
/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages/torch/cuda/__init__.py:143: UserWarning: 
NVIDIA GeForce RTX 3080 Ti with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
If you want to use the NVIDIA GeForce RTX 3080 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))
Net total params: 13.84M
Test DB size: 12.
in infer
Traceback (most recent call last):
  File "infer.py", line 90, in <module>
    main()
  File "infer.py", line 83, in main
    inferNet(infer_data_loader, net, merge_hm_flip_func, merge_tag_flip_func, flip_pairs,
  File "/home2/jaskirat.singh/Drone-based-building-assessment/win_det_heatmaps/common_pytorch/net_modules.py", line 178, in inferNet
    heatmaps, tagmaps = network(batch_data)
  File "/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages/torch/nn/parallel/data_parallel.py", line 166, in forward
    return self.module(*inputs[0], **kwargs[0])
  File "/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "/home2/jaskirat.singh/Drone-based-building-assessment/win_det_heatmaps/common_pytorch/base_modules/architecture.py", line 25, in forward
    x = self.backbone(x)
  File "/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "/home2/jaskirat.singh/Drone-based-building-assessment/win_det_heatmaps/common_pytorch/base_modules/shufflenetv2.py", line 63, in forward
    return self._forward_impl(x)
  File "/home2/jaskirat.singh/Drone-based-building-assessment/win_det_heatmaps/common_pytorch/base_modules/shufflenetv2.py", line 51, in _forward_impl
    x = self.conv1(x)
  File "/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages/torch/nn/modules/container.py", line 141, in forward
    input = module(input)
  File "/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages/torch/nn/modules/activation.py", line 98, in forward
    return F.relu(input, inplace=self.inplace)
  File "/home2/jaskirat.singh/miniconda3/envs/robo/lib/python3.8/site-packages/torch/nn/functional.py", line 1297, in relu
    result = torch.relu_(input)