Closed patrickhwood closed 1 year ago
The weights-20220718T230451Z-001.zip file has a bunch of snunet-*.pt files, but eval.py attempts to load weights/sunet-32.pt.
My patch:
index bd6bbaa..a215ba4 100644 --- a/eval.py +++ b/eval.py @@ -15,7 +15,7 @@ dev = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
test_loader = get_test_loaders(opt)
-path = 'weights/sunet-32.pt' # the path of the model +path = 'weights/snunet-32.pt' # the path of the model model = torch.load(path)
The weights-20220718T230451Z-001.zip file has a bunch of snunet-*.pt files, but eval.py attempts to load weights/sunet-32.pt.
My patch:
index bd6bbaa..a215ba4 100644 --- a/eval.py +++ b/eval.py @@ -15,7 +15,7 @@ dev = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
test_loader = get_test_loaders(opt)
-path = 'weights/sunet-32.pt' # the path of the model +path = 'weights/snunet-32.pt' # the path of the model model = torch.load(path)