wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update lerobot/common/policies/pi0fast/modeling_pi0fast.py 

Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
Dana Aubakirova 2025-04-02 15:03:34 +02:00 committed by GitHub
parent 0b68f2c2fb
commit 2bffbb6e43
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lerobot/common/policies/pi0fast/modeling_pi0fast.py
View File

@ -74,14 +74,6 @@ PRECISION = {
}
def display(tensor: torch.Tensor):
if tensor.dtype == torch.bool:
tensor = tensor.float()
print(f"Shape: {tensor.shape}")
print(f"Mean: {tensor.mean().item()}")
print(f"Std: {tensor.std().item()}")
print(f"Min: {tensor.min().item()}")
print(f"Max: {tensor.max().item()}")
def normalize(x, min_val, max_val):