add log adp objective

kwai · Aug 5, 2021 · e786e21 · e786e21
1 parent d72db82
commit e786e21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/douzero/dmc/arguments.py b/douzero/dmc/arguments.py
@@ -7,7 +7,7 @@
                     help='Experiment id (default: douzero)')
 parser.add_argument('--save_interval', default=30, type=int,
                     help='Time interval (in minutes) at which to save the model')    
-parser.add_argument('--objective', default='adp', type=str, choices=['adp', 'wp'],
+parser.add_argument('--objective', default='adp', type=str, choices=['adp', 'wp', 'logadp'],
                     help='Use ADP or WP as reward (default: ADP)')    
 
 # Training settings