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

Confmat / F1 + Minor fix

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-01-31 16:43:10 -05:00
parent 250ce2c3cf
commit 3ccacd0366
5 changed files with 120 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

2
higher/smart_aug/datasets.py
View file

@ -13,7 +13,7 @@ TEST_SIZE = BATCH_SIZE
#TEST_SIZE = 10000 #legerement +Rapide / + Consomation memoire !
#Wether to download data.
download_data=False
download_data=True
#Number of worker to use.
num_workers=2 #4
#Pin GPU memory