2024/11/14

COMMANDS.md

@@ -1,5 +1,12 @@
 # 常用命令
-
+## 下载数据集
+```zsh
+python extracode/download_dataset/PPG_FieldStudy.py
+```
+## 可视化数据集
+```zsh
+python extracode/visualize/PPG_FieldStudy_pickle.py
+```
 ## 训练
 
 ### WINDOWS+GPU

configs/data/PPG_FieldStudy.yaml

@@ -1,5 +1,5 @@
 _target_: src.data.PPGFieldStudyDatamodule.PPGDataModule
-data_dir: "./data/PPG_FieldStudy"
+data_dir: "./data/PPG_FieldStudy/PPG_FieldStudy"
 window_size: 256
 stride: 128
 batch_size: 64

configs/data/PPG_FieldStudy_MvCFT.yaml

@@ -0,0 +1,9 @@
+_target_: src.data.PPGFieldStudyMvCFTDatamodule.PPGDataModule
+data_dir: "./data/PPG_FieldStudy/PPG_FieldStudy"
+window_size: 256
+stride: 128
+batch_size: 64
+num_workers: 4 # Set to 0 for debugging
+train_split: 0.7
+val_split: 0.15
+test_split: 0.15

configs/experiment/PPG_FieldStudy_MvCFT.yaml

@@ -0,0 +1,41 @@
+# @package _global_
+
+# to execute this experiment run:
+# python train.py experiment=example
+
+defaults:
+  - override /data: PPG_FieldStudy_MvCFT
+  - override /model: PPGMvCFT
+  - override /callbacks: default
+  - override /trainer: default
+
+# all parameters below will be merged with parameters from default configurations set above
+# this allows you to overwrite only specified parameters
+
+tags: ["PPG_FieldStudy", "PPGMvCFT"]
+
+seed: 12345
+
+trainer:
+  min_epochs: 10
+  max_epochs: 50
+  gradient_clip_val: 0.5
+  accelerator: "gpu"
+  devices: 1
+  callbacks: [progress_bar]
+  enable_checkpointing: True
+
+model:
+  input_dim: 14
+  d_model: 128
+  lr: 1e-3
+
+data:
+  batch_size: 64
+
+logger:
+  wandb:
+    tags: ${tags}
+    group: "PPG_FieldStudy"
+  aim:
+    experiment: "PPG_FieldStudy"

configs/model/PPGMvCFT.yaml

@@ -0,0 +1,5 @@
+_target_: src.models.PPGMvCFTModule.PPGMvCFTModule
+
+input_dim: 14
+d_model: 128
+lr: 1e-3