Upload config.yaml

fixed config for the general model

general_2022/config.yaml

@@ -138,7 +138,7 @@ training_settings:
         kernel_size:
         - 9
         - 9
-        adapt_ts: 0.3
+        adapt_ts: 0.2
         scale_factor: 1
         up: true
   stitcher:
@@ -150,15 +150,15 @@ training_settings:
       reduction: mean
   debug_visualiser: null
 model:
-  name: HerdNetTimmDLA
-  from_torchvision: false
+  name: 'HerdNet'
+  from_torchvision: False
   load_from: null
   resume_from: null
   kwargs:
-    backbone: timm/dla34
-    pretrained: true
-    down_ratio: 2
-    head_conv: 64
+    num_layers: 34
+    pretrained: True
+    down_ratio: 2  # was two before
+    head_conv: 64 # was 64
   freeze: null
 wandb_notes: 'fine tune on full iguana dataset with timm DLA34 model