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

Resolution pb memoire apres 1 epoch (Batch size test)

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-01-17 15:12:28 -05:00
parent 5dd0e6ad82
commit b67ec3c469
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
higher/datasets.py
View file

@ -3,8 +3,8 @@ from torch.utils.data import SubsetRandomSampler
import torchvision import torchvision
BATCH_SIZE = 300 BATCH_SIZE = 300
#TEST_SIZE = 300 TEST_SIZE = 300
TEST_SIZE = 10000 #TEST_SIZE = 10000 #legerement +Rapide / + Consomation memoire !
#ATTENTION : Dataug (Kornia) Expect image in the range of [0, 1] #ATTENTION : Dataug (Kornia) Expect image in the range of [0, 1]
#transform_train = torchvision.transforms.Compose([ #transform_train = torchvision.transforms.Compose([
@ -35,7 +35,7 @@ import augmentation_transforms
import numpy as np import numpy as np
download_data=False download_data=False
num_workers=0 num_workers=4 #16
pin_memory=False pin_memory=False
class AugmentedDataset(VisionDataset): class AugmentedDataset(VisionDataset):