hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

update

Browse Source
This commit is contained in:
Thomas Wolf
2024-06-18 13:44:28 +02:00
parent c9150c361b
commit 1cd7ca71a1
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/common/policies/normalize.py
View File

@@ -162,6 +162,9 @@ class Normalize(nn.Module):
output_batch[key] = output_batch[key] * 2 - 1 output_batch[key] = output_batch[key] * 2 - 1
else: else:
raise ValueError(mode) raise ValueError(mode)
for key in batch:
if key not in output_batch:
output_batch[key] = batch[key]
return output_batch return output_batch
@@ -231,4 +234,7 @@ class Unnormalize(nn.Module):
output_batch[key] = output_batch[key] * (max - min) + min output_batch[key] = output_batch[key] * (max - min) + min
else: else:
raise ValueError(mode) raise ValueError(mode)
for key in batch:
if key not in output_batch:
output_batch[key] = batch[key]
return output_batch return output_batch