Spaces:

agueroooooooooo
/

cats_dogs

Runtime error

agueroooooooooo commited on Nov 4, 2022

Commit

27dc96e

1 Parent(s): d6fe440

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ model.to(device)
 criterion = nn.CrossEntropyLoss()
 optimizer = optim.Adam(model.parameters(), lr=0.01)
-checkpoint = torch.load("D:\cats_dogs\cats_dogs\checpoint_epoch_4.pt",
                        map_location=torch.device('cpu'))
 model.load_state_dict(checkpoint["model_state_dict"])
 optimizer.load_state_dict(checkpoint["optimizer_state_dict"])

 criterion = nn.CrossEntropyLoss()
 optimizer = optim.Adam(model.parameters(), lr=0.01)
+checkpoint = torch.load("checpoint_epoch_4.pt",
                        map_location=torch.device('cpu'))
 model.load_state_dict(checkpoint["model_state_dict"])
 optimizer.load_state_dict(checkpoint["optimizer_state_dict"])