skip chunk0: fix ckpt file format

sniperyyc · sniperyyc · commit cec0f3a0658f · 2023-07-26T21:11:52.000-04:00
diff --git a/netshare/model_managers/netshare_manager/netshare_util.py b/netshare/model_managers/netshare_manager/netshare_util.py
@@ -156,7 +156,7 @@ def _configs2configsgroup(
                                 ckpt_dir = os.path.join(
                                     configs[config_id]["result_folder"],
                                     "checkpoint",
-                                    "epoch_id-{}".format(epoch_id)
+                                    "epoch_id-{}.pt".format(epoch_id)
                                 )
                                 if os.path.exists(ckpt_dir):
                                     last_epoch_found = True

Original file line number	Diff line number	Diff line change
`@@ -156,7 +156,7 @@ def _configs2configsgroup(`
`156`	`156`	`ckpt_dir = os.path.join(`
`157`	`157`	`configs[config_id]["result_folder"],`
`158`	`158`	`"checkpoint",`
`159`		`- "epoch_id-{}".format(epoch_id)`
	`159`	`+ "epoch_id-{}.pt".format(epoch_id)`
`160`	`160`	`)`
`161`	`161`	`if os.path.exists(ckpt_dir):`
`162`	`162`	`last_epoch_found = True`