feat: default values, dotenv, gradient clipping, t_max period

DimensionSTP · May 26, 2024 · 6d79521 · 6d79521
1 parent 4d6a30a
commit 6d79521
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/configs/rhythm.yaml b/configs/rhythm.yaml
@@ -9,8 +9,8 @@ defaults:
   - trainer: trainer
 
 package_name: rppg-project
-project_dir: /home/ddang/${package_name}
-connected_dir: /data/${package_name}
+project_dir: ${oc.env:PROJECT_DIR}/${package_name}
+connected_dir: ${oc.env:CONNECTED_DIR}/${package_name}
 
 seed: 2024
 
@@ -31,20 +31,22 @@ rnn_num_layers: 1
 direction: bi
 
 lr: 0.0001
-t_max: 50
-eta_min: 0.000025
+period: 2
+eta_min: 0.00001
 
 monitor: val_rmse_loss
 tracking_direction: min
-patience: 3
+patience: 2
 min_delta: 0
 
-devices: 4
+devices: ${oc.decode:${oc.env:DEVICES}}
 accelerator: gpu
 strategy: ddp
 log_every_n_steps: 10
 precision: 32
 accumulate_grad_batches: 1
+gradient_clip_val: 1
+gradient_clip_algorithm: norm
 epoch: 100
 
 model_name: CustomizedRhythmNet