Spaces:
Running
Running
Commit
·
3822049
1
Parent(s):
72e9451
fix for "latest"
Browse files
vms/ui/project/tabs/train_tab.py
CHANGED
@@ -612,11 +612,13 @@ class TrainTab(BaseTab):
|
|
612 |
has_checkpoints = len(checkpoints) > 0
|
613 |
resume_from = resume_from_checkpoint # Use the passed parameter
|
614 |
|
615 |
-
if resume_from
|
616 |
# Find the latest checkpoint
|
617 |
latest_checkpoint = max(checkpoints, key=os.path.getmtime)
|
618 |
resume_from = str(latest_checkpoint)
|
619 |
-
|
|
|
|
|
620 |
|
621 |
# Convert model_type display name to internal name
|
622 |
model_internal_type = MODEL_TYPES.get(model_type)
|
|
|
612 |
has_checkpoints = len(checkpoints) > 0
|
613 |
resume_from = resume_from_checkpoint # Use the passed parameter
|
614 |
|
615 |
+
if resume_from and checkpoints:
|
616 |
# Find the latest checkpoint
|
617 |
latest_checkpoint = max(checkpoints, key=os.path.getmtime)
|
618 |
resume_from = str(latest_checkpoint)
|
619 |
+
|
620 |
+
logger.info(f"Found checkpoint at {resume_from}, note from @julian: right now let's just resume training at 'latest'")
|
621 |
+
result_from = "latest"
|
622 |
|
623 |
# Convert model_type display name to internal name
|
624 |
model_internal_type = MODEL_TYPES.get(model_type)
|