From 629fc212d4a69dd938fc7735e58df227119c75b2 Mon Sep 17 00:00:00 2001 From: Ghassen Date: Tue, 2 Jan 2024 10:46:04 +0100 Subject: [PATCH] Update resnet18 model path --- src/dot/simswap/parsing_model/resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/dot/simswap/parsing_model/resnet.py b/src/dot/simswap/parsing_model/resnet.py index 1ecc4af..9786a6b 100644 --- a/src/dot/simswap/parsing_model/resnet.py +++ b/src/dot/simswap/parsing_model/resnet.py @@ -6,7 +6,7 @@ import torch.nn.functional as F import torch.utils.model_zoo as modelzoo -resnet18_url = "https://download.pytorch.org/models/resnet18-5c106cde.pth" +resnet18_url = "saved_models/simswap/resnet18-5c106cde.pth" def conv3x3(in_planes, out_planes, stride=1):