添加sttn训练代码

2026-05-21 05:24:51 +08:00 · 2024-01-08 17:47:59 +08:00
parent 4abc3409ac
commit d6736d9206
11 changed files with 848 additions and 3 deletions
--- a/backend/tools/train/configs_sttn/davis.json
+++ b/backend/tools/train/configs_sttn/davis.json
@@ -0,0 +1,33 @@
+{
+    "seed": 2020,
+    "save_dir": "release_model/",
+    "data_loader": {
+        "name": "davis",
+        "data_root": "datasets/",
+        "w": 432,
+        "h": 240,
+        "sample_length": 5
+    },
+    "losses": {
+        "hole_weight": 1,
+        "valid_weight": 1,
+        "adversarial_weight": 0.01,
+        "GAN_LOSS": "hinge"
+    },
+    "trainer": {
+        "type": "Adam",
+        "beta1": 0,
+        "beta2": 0.99,
+        "lr": 1e-4,
+        "d2glr": 1, 
+        "batch_size": 8,
+        "num_workers": 2,
+        "verbosity": 2,
+        "log_step": 100,
+        "save_freq": 1e4,
+        "valid_freq": 1e4, 
+        "iterations": 50e4,
+        "niter": 30e4,
+        "niter_steady": 30e4
+    }
+}