Fix bug with wrong argument order

2021-07-29 15:30:55 +02:00
parent 3b623a1467
commit afc3719ab9
1 changed files with 1 additions and 1 deletions
--- a/src/main.py
+++ b/src/main.py
@@ -54,7 +54,7 @@ def main(config, method='bc', train=False, test=False, loc=0, datadir='./expert_
    if test:

        # load policy 
-        policy = policy_class.load_model(config, filestr)
+        policy = policy_class.load_model(filestr, config)
        policy.eval()

        # simulate policy