Skip to content

GitLab

Commit 3c3b0f8e authored by Andrea Paris's avatar Andrea Paris
Browse files

cleanup

parent e6b5a952
Show whitespace changes
Inline Side-by-side
Showing with 9 additions and 9 deletions
examples/depth/train.py
View file @ 3c3b0f8e
...@@ -437,16 +437,16 @@ def main( ...@@ -437,16 +437,16 @@ def main(
# specify which models to train here # specify which models to train here
models = [ models = [
#"transformer_sc2_layers4_e128", "transformer_sc2_layers4_e128",
#"s2transformer_sc2_layers4_e128", "s2transformer_sc2_layers4_e128",
#"ntransformer_sc2_layers4_e128", "ntransformer_sc2_layers4_e128",
"s2ntransformer_sc2_layers4_e128", "s2ntransformer_sc2_layers4_e128",
#"segformer_sc2_layers4_e128", "segformer_sc2_layers4_e128",
#"s2segformer_sc2_layers4_e128", "s2segformer_sc2_layers4_e128",
#"nsegformer_sc2_layers4_e128", "nsegformer_sc2_layers4_e128",
#"s2nsegformer_sc2_layers4_e128", "s2nsegformer_sc2_layers4_e128",
#"sfno_sc2_layers4_e32", "sfno_sc2_layers4_e32",
#"lsno_sc2_layers4_e32", "lsno_sc2_layers4_e32",
] ]
models = {k: baseline_models[k] for k in models} models = {k: baseline_models[k] for k in models}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment