TypeError: __init__() got an unexpected keyword argument 'weights'

[Nick Pai]

Has anyone solved the issue when trying to run the RLT/main.py encounter TypeError: __init__() got an unexpected keyword argument 'weights' ? Or any suggestion to walk around this issue?

root@6d952ea9a8d1:/monoport# python RTL/main.py --use_server --ip 192.168.0.103 --port 5555 --image_folder images -- netG.ckpt_path ./data/PIFu/net_G netC.ckpt_path ./data/PIFu/net_C
loading networkG from ./data/PIFu/net_G ...
loading networkC from ./data/PIFu/net_C ...
initialize data streamer ...
Using cache found in /root/.cache/torch/hub/NVIDIA_DeepLearningExamples_torchhub
/root/.cache/torch/hub/NVIDIA_DeepLearningExamples_torchhub/PyTorch/Classification/ConvNets/image_classification/models/efficientnet.py:18: UserWarning: pytorch_quantization module not found, quantization will not be available
  "pytorch_quantization module not found, quantization will not be available"
Using cache found in /root/.cache/torch/hub/NVIDIA_DeepLearningExamples_torchhub
Traceback (most recent call last):
  File "RTL/main.py", line 157, in <module>
    device=cuda_backbone_G, verbose=False)
  File "/monoport/human_inst_seg/human_inst_seg/unet.py", line 64, in __init__
    self.det_engine = Detection(device=device)
  File "/monoport/human_det/human_det/ssd.py", line 23, in __init__
    weights='IMAGENET1K_V1'
  File "/usr/local/lib/python3.7/site-packages/torch/hub.py", line 339, in load
    model = _load_local(repo_or_dir, model, *args, **kwargs)
  File "/usr/local/lib/python3.7/site-packages/torch/hub.py", line 368, in _load_local
    model = entry(*args, **kwargs)
  File "/root/.cache/torch/hub/NVIDIA_DeepLearningExamples_torchhub/PyTorch/Detection/SSD/ssd/entrypoints.py", line 179, in nvidia_ssd
    from . import model as ssd
  File "/root/.cache/torch/hub/NVIDIA_DeepLearningExamples_torchhub/PyTorch/Detection/SSD/ssd/model.py", line 55, in <module>
    class SSD300(nn.Module):
  File "/root/.cache/torch/hub/NVIDIA_DeepLearningExamples_torchhub/PyTorch/Detection/SSD/ssd/model.py", line 56, in SSD300
    def __init__(self, backbone=ResNet('resnet50')):
  File "/root/.cache/torch/hub/NVIDIA_DeepLearningExamples_torchhub/PyTorch/Detection/SSD/ssd/model.py", line 30, in __init__
    backbone = resnet50(weights=None if backbone_path else weights)
  File "/usr/local/lib/python3.7/site-packages/torchvision/models/resnet.py", line 301, in resnet50
    **kwargs)
  File "/usr/local/lib/python3.7/site-packages/torchvision/models/resnet.py", line 260, in _resnet
    model = ResNet(block, layers, **kwargs)
TypeError: __init__() got an unexpected keyword argument 'weights'