/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
=================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)
