(train)
| 367 | |
| 368 | |
| 369 | def get_transform(train): |
| 370 | transforms = [] |
| 371 | if train: |
| 372 | transforms.append(T.RandomHorizontalFlip(0.5)) |
| 373 | transforms.append(T.ToDtype(torch.float, scale=True)) |
| 374 | transforms.append(T.ToPureTensor()) |
| 375 | return T.Compose(transforms) |
| 376 | |
| 377 | ###################################################################### |
| 378 | # Testing ``forward()`` method (Optional) |
no outgoing calls
no test coverage detected