22 lines
367 B
YAML
22 lines
367 B
YAML
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
|
|
|