Skip to content

Commit

Permalink
Change run id format to avoid name collisions
Browse files Browse the repository at this point in the history
  • Loading branch information
joeloskarsson committed Nov 25, 2023
1 parent 9912ece commit cd94f57
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion train_model.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
import random
import torch
import pytorch_lightning as pl
from lightning_fabric.utilities import seed
Expand Down Expand Up @@ -82,6 +83,9 @@ def main():
assert args.step_length <= 3, "Too high step length"
assert args.eval in (None, "val", "test"), f"Unknown eval setting: {args.eval}"

# Get an (actual) random run id as a unique identifier
random_run_id = random.randint(0, 9999)

# Set seed
seed.seed_everything(args.seed)

Expand Down Expand Up @@ -120,7 +124,7 @@ def main():
if args.eval:
prefix = prefix + f"eval-{args.eval}-"
run_name = f"{prefix}{args.model}-{args.processor_layers}x{args.hidden_dim}-"\
f"{time.strftime('%m_%d_%H_%M_%S')}"
f"{time.strftime('%m_%d_%H')}-{random_run_id:04d}"
checkpoint_callback = pl.callbacks.ModelCheckpoint(
dirpath=f"saved_models/{run_name}", filename="min_val_loss",
monitor="val_mean_loss", mode="min", save_last=True)
Expand Down

0 comments on commit cd94f57

Please sign in to comment.