Skip to content

Commit 4e791de

Browse files
committed
update hw3_helper
1 parent 7aeffac commit 4e791de

File tree

2 files changed

+1
-1
lines changed

2 files changed

+1
-1
lines changed

deepul/hw3_helper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def calculate_is(samples):
8080
assert (len(samples[0].shape) == 3)
8181

8282
model = GoogLeNet().to(ptu.device)
83-
model.load_state_dict(torch.load("deepul/deepul/hw4_utils/classifier.pt"))
83+
model.load_state_dict(torch.load("deepul/deepul/hw3_utils/classifier.pt"))
8484
softmax = nn.Sequential(model, nn.Softmax(dim=1))
8585

8686
bs = 100

deepul/hw3_utils/lena.jpg

102 KB
Loading

0 commit comments

Comments
 (0)