initial commit

2021-08-05 11:59:22 +08:00
parent a18b0197d8
commit 791f02f280
10 changed files with 275 additions and 6 deletions
--- a/fit/configs/config.json
+++ b/fit/configs/config.json
@ -0,0 +1,25 @@
+{
+    "MODEL": {
+        "GENDER": "male"
+    },
+    "TRAIN": {
+        "LEARNING_RATE":2e-2,
+        "MAX_EPOCH": 5,
+        "WRITE": 1,
+        "SAVE": 10,
+        "BATCH_SIZE": 1,
+        "MOMENTUM": 0.9,
+        "lr_scheduler": {
+            "T_0": 10,
+            "T_mult": 2,
+            "eta_min": 1e-2
+        },
+        "loss_func": ""
+    },
+    "USE_GPU": 1,
+    "DATA_LOADER": {
+        "NUM_WORKERS": 1
+    },
+    "TARGET_PATH":"../Action2Motion/HumanAct12/HumanAct12/P01G01R01F0069T0143A0102.npy",
+    "DATASET_PATH":"../Action2Motion/HumanAct12/HumanAct12/"
+}