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

Remove prints

This commit is contained in:
Simon Alibert 2024-06-05 12:21:00 +00:00
parent 6509c3f6d4
commit ceb95592af
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/transforms.py
View File

@ -57,7 +57,6 @@ class RandomSubsetApply(Transform):
selected_indices = selected_indices.sort().values selected_indices = selected_indices.sort().values
selected_transforms = [self.transforms[i] for i in selected_indices] selected_transforms = [self.transforms[i] for i in selected_indices]
print(selected_transforms)
for transform in selected_transforms: for transform in selected_transforms:
outputs = transform(*inputs) outputs = transform(*inputs)
@ -93,7 +92,6 @@ class RangeRandomSharpness(Transform):
def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any: def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
sharpness_factor = self.range_min + (self.range_max - self.range_min) * torch.rand(1).item() sharpness_factor = self.range_min + (self.range_max - self.range_min) * torch.rand(1).item()
print(f"{sharpness_factor=}")
return self._call_kernel(F.adjust_sharpness, inpt, sharpness_factor=sharpness_factor) return self._call_kernel(F.adjust_sharpness, inpt, sharpness_factor=sharpness_factor)