-
Notifications
You must be signed in to change notification settings - Fork 27
Description
Hi,
I'm trying to test out the pre-trained model and am having issues downloading the state dictionary with the "main.py" script. Here is the error I'm getting:
line 461, in load_state_dict_from_url
hash_prefix = HASH_REGEX.search(filename).group(1)
AttributeError: 'NoneType' object has no attribute 'group'
I've tried updating PyTorch and Torchvision to the latest releases, but got the same error. I then downloaded the state dictionary manually from the provided URL and using torch.load to load it into the variable "state_dict", but ran into more issues when trying to load that with the model. I've been using the following command to run the script:
python main.py [train and validation folders] --arch vovnet39 --pretrained --savedir VoVNet39
Thanks,
Bryan