training updates

This commit is contained in:
Oliver 2025-04-28 20:56:23 +01:00
parent e58f3c1a35
commit 55c9810113
2 changed files with 11 additions and 21 deletions

View file

@ -171,7 +171,7 @@ trainer:
logger: [ 'console', 'wandb' ]
val_generations_to_log_to_wandb: 0
nnodes: 1
n_gpus_per_node: 4
n_gpus_per_node: 2
save_freq: 100
# auto: find the last ckpt to resume. If can't find, start from scratch
resume_mode: auto # or auto or resume_path if