diff --git a/configs/rhythm.yaml b/configs/rhythm.yaml index a765204..ad283ba 100644 --- 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