code

svip-lab · Jun 16, 2021 · 4b66f14 · 4b66f14
1 parent daadb28
commit 4b66f14
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train_agent.py b/train_agent.py
@@ -95,6 +95,7 @@ def davis_config(run, _log):
     davis.sets[subset] = seq_list
     cfg_yl.setting = 'oracle'
     cfg_yl.method = 'ours'
+    set_random_seed(2019)
     cfg_yl.num_epochs = 5
     report_save_dir = save_result_dir