Skip to content

Commit 521fd65

Browse files
authored
Update demo_dataloader.py
Removed out-dated comments.
1 parent 78a8522 commit 521fd65

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

demo_dataloader.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,10 +68,6 @@ def create_demo_dataset(self, num_images=1001):
6868

6969
@staticmethod
7070
def get_transforms(image_size, use_tensordict):
71-
# returns 3 transforms: norm, main, and to_tensor.
72-
# norm: normalizes image and mask.
73-
# main: resize (and rotate, flip, etc. if not use_tensordict) for image and mask.
74-
# to_tensor: for image and mask.
7571
if use_tensordict:
7672
# when using tensordict memory mapping, probabilities should not be included in Dataset.
7773
# probability based transforms must be moved from Dataset into Collate_Fn.

0 commit comments

Comments
 (0)