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

Add comment

Browse Source
This commit is contained in:
Johannes Fischer
2021-07-30 18:19:29 +02:00
parent 367f72ec46
commit 99f7df2e7c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/metrics.py
View File

@@ -18,7 +18,7 @@ def metrics(filestr: str, test_dataset, policy):
# a) simulation files that were saved under the trained policy with prefix 'policy' # a) simulation files that were saved under the trained policy with prefix 'policy'
# b) applying the policy to observations in the test dataset # b) applying the policy to observations in the test dataset
# load trajectory # load simulated trajectory
states = torch.load(filestr + '_sim_states.pt').detach() states = torch.load(filestr + '_sim_states.pt').detach()
lengths = torch.load(filestr + '_sim_lengths.pt').detach() lengths = torch.load(filestr + '_sim_lengths.pt').detach()
widths = torch.load(filestr + '_sim_widths.pt').detach() widths = torch.load(filestr + '_sim_widths.pt').detach()