Skip to content

Instantly share code, notes, and snippets.

@tkshnkmr

tkshnkmr/my_get_transform.py

Last active September 28, 2019 14:05
Show Gist options
  • Save tkshnkmr/6eae47e1227ec19a3e4d83eb2bae8437 to your computer and use it in GitHub Desktop.
Save tkshnkmr/6eae47e1227ec19a3e4d83eb2bae8437 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
my_get_transform.py
# In my case, just added ToTensor
def get_transform():
custom_transforms = []
custom_transforms.append(torchvision.transforms.ToTensor())
return torchvision.transforms.Compose(custom_transforms)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment