configs/trainer/default.yaml

default_root_dir: ${paths.output_dir}

max_epochs: 5

accelerator: gpu
devices: 1

# mixed precision for extra speed-up
precision: "32-true"

# perform a validation loop every N training epochs
# check_val_every_n_epoch: 1

log_interval: 1
print_interval: 10
save_ckpt: all

fabric:
  _target_: lightning.Fabric
  accelerator: ${trainer.accelerator}
  devices: ${trainer.devices}
  precision: ${trainer.precision}
  loggers: ${trainer.logger}