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 cast in Augmented Dataset

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-12-04 12:58:11 -05:00
parent 2ee8022c2f
commit adaac437b6
3 changed files with 29 additions and 822 deletions
Download patch file Download diff file Expand all files Collapse all files

5
higher/test_dataug.py
View file

@ -69,7 +69,7 @@ if __name__ == "__main__":
'aug_dataset',
#'aug_model'
}
n_inner_iter = 0
n_inner_iter = 1
epochs = 100
dataug_epoch_start=0
@ -104,6 +104,9 @@ if __name__ == "__main__":
#### Augmented Dataset ####
if 'aug_dataset' in tasks:
xs, ys = next(iter(dl_train))
viz_sample_data(imgs=xs, labels=ys, fig_name='samples/data_sample_{}'.format(str(data_train_aug)))
t0 = time.process_time()
model = LeNet(3,10).to(device)
#model = WideResNet(num_classes=10, wrn_size=16).to(device)