model:
  model_class: model.pod_nn.PODNN
  pod_rank: 128
  layers: [16, 64, 128, 128, 64, 16]
   
optimizer:
  optimizer_class: torch.optim.AdamW
  lr: 0.001

trainer: 
  max_epochs: 10000
  batch_size: 128
  train_size: 0.9
  val_size: 0.1
  accelerator: 'cuda'
  devices: 1
  log_every_n_steps: 0
  patience: 100
  log_dir: logs/
  name: pod_nn
  # version: null