DouZero
/
Douzero_Resnet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

参数调整

This commit is contained in:
ZaneYork 2021-12-05 13:01:29 +08:00
parent 56c6ac5130
commit e017c3724d
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
douzero/dmc/arguments.py
View File

@ -5,7 +5,7 @@ parser = argparse.ArgumentParser(description='DouZero: PyTorch DouDizhu AI')
# General Settings # General Settings
parser.add_argument('--xpid', default='douzero', parser.add_argument('--xpid', default='douzero',
help='Experiment id (default: douzero)') help='Experiment id (default: douzero)')
parser.add_argument('--save_interval', default=10, type=int, parser.add_argument('--save_interval', default=30, type=int,
help='Time interval (in minutes) at which to save the model') help='Time interval (in minutes) at which to save the model')
parser.add_argument('--objective', default='adp', type=str, choices=['adp', 'wp', 'logadp'], parser.add_argument('--objective', default='adp', type=str, choices=['adp', 'wp', 'logadp'],
help='Use ADP or WP as reward (default: ADP)') help='Use ADP or WP as reward (default: ADP)')

1
train.cmd Normal file
View File

@ -0,0 +1 @@
python3 train.py --actor_device_cpu --training_device cpu --load_model