feat: store metric logs inside each checkpoint dir, remove log_dir config
This commit is contained in:
@@ -39,7 +39,6 @@ def create_train_config(
|
||||
optimizer_fn=optimizer_fn,
|
||||
scheduler_fn=scheduler_fn,
|
||||
ckpt_dir=test_dir,
|
||||
log_dir=os.path.join(test_dir, "logs"),
|
||||
n_epoch=n_epoch,
|
||||
batch_per_device=batch_per_device,
|
||||
ckpt_interval=ckpt_interval,
|
||||
|
||||
Reference in New Issue
Block a user