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

Fix etat Train/Eval pour augmentation differee (Retester !)

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-01-20 17:09:31 -05:00
parent 2d6d2f7397
commit d21a6bbf5c
4 changed files with 38 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

5
higher/test_dataug.py
View file

@ -68,7 +68,7 @@ if __name__ == "__main__":
}
n_inner_iter = 1
epochs = 150
dataug_epoch_start=0
dataug_epoch_start=10
optim_param={
'Meta':{
'optim':'Adam',
@ -87,8 +87,6 @@ if __name__ == "__main__":
#model = MobileNetV2(num_classes=10)
#model = WideResNet(num_classes=10, wrn_size=32)
model = Higher_model(model) #run_dist_dataugV3
#### Classic ####
if 'classic' in tasks:
t0 = time.process_time()
@ -171,6 +169,7 @@ if __name__ == "__main__":
t0 = time.process_time()
tf_dict = {k: TF.TF_dict[k] for k in tf_names}
model = Higher_model(model) #run_dist_dataugV3
aug_model = Augmented_model(Data_augV5(TF_dict=tf_dict, N_TF=2, mix_dist=0.8, fixed_prob=False, fixed_mag=False, shared_mag=False), model).to(device)
#aug_model = Augmented_model(RandAug(TF_dict=tf_dict, N_TF=2), model).to(device)