Skip to content

F
FML Project
Commit 9d6946db authored by Meet Narendra's avatar Meet Narendra 💬
Browse files
Options

Modified batch size

parent c4288d29
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
1703.10593/preprocess.py
View file @ 9d6946db
...@@ -17,8 +17,8 @@ class LoadData(Dataset): ...@@ -17,8 +17,8 @@ class LoadData(Dataset):
self.pair = pair self.pair = pair
self.type = type_data self.type = type_data
self.transform = transforms.Compose([ self.transform = transforms.Compose([
transforms.Resize(int(64 * 1.12), Image.BICUBIC), transforms.Resize(int(256 * 1.12), Image.BICUBIC),
transforms.RandomCrop(64), transforms.RandomCrop(256),
transforms.RandomHorizontalFlip(), transforms.RandomHorizontalFlip(),
transforms.ToTensor(), transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment