Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # In my case, just added ToTensor
- def get_transform():
- custom_transforms = []
- custom_transforms.append(torchvision.transforms.ToTensor())
- return torchvision.transforms.Compose(custom_transforms)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement