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

Test WRN Brutus

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-12-04 16:34:02 -05:00
parent b26fbcd2a2
commit fa5bc72616
6 changed files with 72 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

6
higher/train_utils.py
View file

@ -625,9 +625,9 @@ def run_dist_dataugV2(model, epochs=1, inner_it=0, dataug_epoch_start=0, print_f
model_copy(src=fmodel, dst=model)
optim_copy(dopt=diffopt, opt=inner_opt)
if epoch>50:
meta_opt.step()
model['data_aug'].adjust_param(soft=False) #Contrainte sum(proba)=1
#if epoch>50:
meta_opt.step()
model['data_aug'].adjust_param(soft=False) #Contrainte sum(proba)=1
#model['data_aug'].next_TF_set()
fmodel = higher.patch.monkeypatch(model, device=None, copy_initial_weights=True)