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/experiments/pod_nn.yaml
+++ b/experiments/pod_nn.yaml
@@ -1,22 +0,0 @@
-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
-