Error with resnet in jupyter notebook

[reillydonovan]

Hi, when I try to run section 21 of the jupyter notebook I am returned the follow error: Downloading resnet50_places365.pth.tar from http://places2.csail.mit.edu/models_places365/resnet50_places365.pth.tar it may take some time.

---

NameError Traceback (most recent call last) /tmp/ipykernel_212303/229011906.py in 10 config['spatial_shift_size'] = 40 11 ---> 12 img = deep_dream_static_image(config) 13 dump_path = save_and_maybe_display_image(config, img) 14 print(f'Saved DeepDream static image to: {os.path.relpath(dump_path)}\n')

/tmp/ipykernel_212303/1103483755.py in deep_dream_static_image(config, img) 1 def deep_dream_static_image(config, img=None): ----> 2 model = fetch_and_prepare_model(config['model_name'], config['pretrained_weights']) 3 4 try: 5 layer_ids_to_use = [model.layer_names.index(layer_name) for layer_name in config['layers_to_use']]

/tmp/ipykernel_212303/3540351205.py in fetch_and_prepare_model(model_type, pretrained_weights) 128 elif model_type == SupportedModels.RESNET50.name: 129 # We'll define the ResNet50 later --> 130 model = ResNet50(pretrained_weights, requires_grad=False, show_progress=True).to(DEVICE) 131 else: 132 raise Exception('Model not yet supported.')

/tmp/ipykernel_212303/2427502766.py in init(self, pretrained_weights, requires_grad, show_progress) 17 binary_url = r'http://places2.csail.mit.edu/models_places365/resnet50_places365.pth.tar' 18 print(f'Downloading {binary_name} from {binary_url} it may take some time.') ---> 19 download_url_to_file(binary_url, resnet50_places365_binary_path) 20 print('Done downloading.') 21 state_dict = torch.load(resnet50_places365_binary_path)['state_dict']

NameError: name 'download_url_to_file' is not defined

Any idea how to fix this? Thanks!

[afruehstueck]

You need to import that function from torch.hub: from torch.hub import download_url_to_file