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

Merge branch 'main' of https://github.com/sisl/InteractionImitation into main

Browse Source
This commit is contained in:
Arec
2021-07-23 04:44:57 -07:00
parent 5b09374c77 8a4094037a
commit 969812c5cc
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/policies/test_forward_pass.py
View File

@@ -10,8 +10,12 @@ with open(config_path, 'r') as cfg:
filestr = 'tests/policies/base_' filestr = 'tests/policies/base_'
batch = pickle.load(open(filestr+'_test_batch.pkl', 'rb')) def test_forward_pass():
batch = pickle.load(open(filestr + '_test_batch.pkl', 'rb'))
batch['relative_state'] = batch['relative_state'].float() batch['relative_state'] = batch['relative_state'].float()
policy = BehaviorCloningPolicy.load_model(config, filestr) policy = BehaviorCloningPolicy.load_model(config, filestr)
policy.policy = policy.policy.type(batch['state'].dtype) policy.policy = policy.policy.type(batch['state'].dtype)
action = policy(batch) action = policy(batch)
i = torch.where(action.isnan())[0]
for key in batch.keys():
print(batch[key][i])