[Enhancement] Add sample-model checking for evaluation (#165)

open-mmlab · Nov 29, 2021 · bf75868 · bf75868
1 parent 2ab33f1
commit bf75868
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/evaluation.py b/tools/evaluation.py
@@ -54,6 +54,7 @@ def parse_args():
         '--sample-model',
         type=str,
         default='ema',
+        choices=['ema', 'orig'],
         help='use which mode (ema/orig) in sampling')
     parser.add_argument(
         '--eval',