gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

Adding parameteres, checking transitions before training

Browse Source
This commit is contained in:
Ajay Deshpande
2018-10-08 14:49:46 -07:00
committed by zach dwiel
parent 0f46877d7e
commit b285a02023
5 changed files with 18 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
rl_coach/training_worker.py
View File

@@ -48,22 +48,14 @@ def main():
help="(string) Name of a preset to run (class name from the 'presets' directory.)",
type=str,
required=True)
parser.add_argument('--checkpoint_dir',
parser.add_argument('--checkpoint-dir',
help='(string) Path to a folder containing a checkpoint to write the model to.',
type=str,
default='/checkpoint')
parser.add_argument('-r', '--redis_ip',
help="(string) IP or host for the redis server",
default='localhost',
type=str)
parser.add_argument('-rp', '--redis_port',
help="(int) Port of the redis server",
default=6379,
type=int)
parser.add_argument('--memory_backend_params',
parser.add_argument('--memory-backend-params',
help="(string) JSON string of the memory backend params",
type=str)
parser.add_argument('--data_store_params',
parser.add_argument('--data-store-params',
help="(string) JSON string of the data store params",
type=str)
args = parser.parse_args()