Add testing script for raytune experiments

2021-07-29 15:30:41 +02:00
parent 35634fd2eb
commit 3b623a1467
2 changed files with 14 additions and 6 deletions
--- a/src/main.py
+++ b/src/main.py
@@ -27,7 +27,7 @@ def main(config, method='bc', train=False, test=False, loc=0, datadir='./expert_
        test (bool): whether to run test loop
        method (str): the method to try for imitation
        loc (int): the location index of the roundabout
-        config_file (str): path to config file
+        datadir (str): path to expert data
        kwargs (dict): remaining kwargs for training loop
    """
    # get/set seed