Skip to content
Snippets Groups Projects
Commit 305802f6 authored by Dipam Chakraborty's avatar Dipam Chakraborty
Browse files

No LR Scheduler + 10 Epochs + Unfreeze all layers

parent 7ecb50b6
No related branches found
No related tags found
No related merge requests found
......@@ -96,6 +96,7 @@ class ZEWDPCTrainer:
T.ToTensor(), # Converts image to [0, 1]
T.RandomVerticalFlip(p=0.5),
T.RandomHorizontalFlip(p=0.5),
T.GaussianBlur(kernel_size=3),
T.ColorJitter(brightness=0.2, contrast=0.2),
# TODO: Could agree with participants on the final list of transforms
*self.model.required_transforms,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment