Skip to content

Commit e4cc2b9

Browse files
committed
changed minor path dir
1 parent 2bad8ab commit e4cc2b9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

app.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def get_device() -> torch.device:
3434

3535
# initialize G (young→old) and F (old→young)
3636
G, F, _, _ = initialize_models()
37-
ckpt_path = Path("outputs/checkpoints/epoch_0030.pth")
37+
ckpt_path = Path("outputs/checkpoints/best.pth")
3838
ckpt = torch.load(ckpt_path, map_location=device)
3939

4040
G.load_state_dict(ckpt["G"])

0 commit comments

Comments
 (0)