Skip to content

Commit 9e500a6

Browse files
authored
Update googlenet_places365.py
1 parent 9cfd1b8 commit 9e500a6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

googlenet_places365.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -265,8 +265,8 @@ def forward(self, x):
265265
return pool5_drop_7x7_s1#, loss2_classifier_1, loss1_classifier_1
266266

267267

268-
cnn = GoogLeNetPlaces205()
269-
cnn.load_state_dict(torch.load("googlenet_places205.pth"))
268+
cnn = GoogLeNetPlaces365()
269+
cnn.load_state_dict(torch.load("googlenet_places365.pth"))
270270
cnn.eval()
271271

272272
print(cnn)

0 commit comments

Comments
 (0)