AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-05-04 12:10:45 +02:00
Code Issues Projects Releases Packages Wiki Activity

LR scheduler + Resolution pb ResNet50/WRN

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-02-12 13:43:44 -05:00
parent 383f63c7b8
commit 79de0191a8
4 changed files with 78 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

2
higher/smart_aug/datasets.py
View file

@ -7,7 +7,7 @@ from torch.utils.data.dataset import ConcatDataset
import torchvision
#Train/Validation batch size.
BATCH_SIZE = 300
BATCH_SIZE = 512
#Test batch size.
TEST_SIZE = BATCH_SIZE
#TEST_SIZE = 10000 #legerement +Rapide / + Consomation memoire !