cleanup

3c3b0f8e · Andrea Paris · e6b5a952 · 3c3b0f8e
Commit 3c3b0f8e authored Jul 01, 2025 by Andrea Paris
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 9 deletions

examples/depth/train.py examples/depth/train.py +9 -9

No files found.
--- a/examples/depth/train.py
+++ b/examples/depth/train.py
@@ -437,16 +437,16 @@ def main(

    # specify which models to train here
    models = [
-        #"transformer_sc2_layers4_e128",
-        #"s2transformer_sc2_layers4_e128",
-        #"ntransformer_sc2_layers4_e128",
+        "transformer_sc2_layers4_e128",
+        "s2transformer_sc2_layers4_e128",
+        "ntransformer_sc2_layers4_e128",
        "s2ntransformer_sc2_layers4_e128",
-        #"segformer_sc2_layers4_e128",
-        #"s2segformer_sc2_layers4_e128",
-        #"nsegformer_sc2_layers4_e128",
-        #"s2nsegformer_sc2_layers4_e128",
-        #"sfno_sc2_layers4_e32",
-        #"lsno_sc2_layers4_e32",
+        "segformer_sc2_layers4_e128",
+        "s2segformer_sc2_layers4_e128",
+        "nsegformer_sc2_layers4_e128",
+        "s2nsegformer_sc2_layers4_e128",
+        "sfno_sc2_layers4_e32",
+        "lsno_sc2_layers4_e32",
    ]
    models = {k: baseline_models[k] for k in models}