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

Remove prints

Browse Source
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
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)