CONTA
CONTA copied to clipboard
_pickle.UnpicklingError: invalid load key
When attempting to execute main.py a pickling error is returned and after tracing it back it is state_dict = torch.load(CONFIG.MODEL.INIT_MODEL)
line that is throwing the error. Also, the INIT_MODEL in the voc12.yaml file is a pretrained resnet model. The versions of torch and torchvision match those in the requirements. Wondering if this error is due to potentially using different versions of Python? Any suggestions?