From 8b45a93691f39641ecb24b295edc62d9116c8878 Mon Sep 17 00:00:00 2001 From: ProGamerGov Date: Tue, 17 May 2022 18:17:02 -0600 Subject: [PATCH] Change Inception5h model link --- captum/optim/models/_image/inception_v1.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/captum/optim/models/_image/inception_v1.py b/captum/optim/models/_image/inception_v1.py index 682da207f6..d24c87d42d 100644 --- a/captum/optim/models/_image/inception_v1.py +++ b/captum/optim/models/_image/inception_v1.py @@ -5,10 +5,7 @@ import torch.nn as nn from captum.optim.models._common import Conv2dSame, RedirectedReluLayer, SkipLayer -GS_SAVED_WEIGHTS_URL = ( - "https://github.com/pytorch/captum/raw/" - + "optim-wip/captum/optim/models/_image/inception5h.pth" -) +GS_SAVED_WEIGHTS_URL = "https://pytorch.s3.amazonaws.com/models/captum/inception5h.pth" def googlenet(