move POD-NN model

2025-10-02 10:20:16 +02:00
parent 4253469ba7
commit 876efea0a6
2 changed files with 0 additions and 0 deletions
--- a/pod_nn/experiments/pod_nn.yaml
+++ b/pod_nn/experiments/pod_nn.yaml
@@ -0,0 +1,22 @@
+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
+