Skip to content

Commit f2b6c26

Browse files
author
Arthur Douillard
committed
[data] Add brighness aug to cifar100.
1 parent f7b344b commit f2b6c26

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

inclearn/data.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,8 @@ class iCIFAR10(IncrementalDataset):
136136
_base_dataset = datasets.cifar.CIFAR10
137137
_train_transforms = [
138138
transforms.RandomCrop(32, padding=4),
139-
transforms.RandomHorizontalFlip()
139+
transforms.RandomHorizontalFlip(),
140+
transforms.ColorJitter(brightness=63 / 255)
140141
]
141142
_common_transforms = [
142143
transforms.ToTensor(),

0 commit comments

Comments
 (0)