huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix bug with wrong argument order

Browse Source
This commit is contained in:
Johannes Fischer
2021-07-29 15:30:55 +02:00
parent 3b623a1467
commit afc3719ab9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main.py
View File

@@ -54,7 +54,7 @@ def main(config, method='bc', train=False, test=False, loc=0, datadir='./expert_
if test: if test:
# load policy # load policy
policy = policy_class.load_model(config, filestr) policy = policy_class.load_model(filestr, config)
policy.eval() policy.eval()
# simulate policy # simulate policy