We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2bad8ab commit e4cc2b9Copy full SHA for e4cc2b9
app.py
@@ -34,7 +34,7 @@ def get_device() -> torch.device:
34
35
# initialize G (young→old) and F (old→young)
36
G, F, _, _ = initialize_models()
37
-ckpt_path = Path("outputs/checkpoints/epoch_0030.pth")
+ckpt_path = Path("outputs/checkpoints/best.pth")
38
ckpt = torch.load(ckpt_path, map_location=device)
39
40
G.load_state_dict(ckpt["G"])
0 commit comments