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 KL divergence from UDA

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-12-06 10:44:18 -05:00
parent fa5bc72616
commit 217f94ef89
5 changed files with 52 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

1
higher/datasets.py
View file

@ -50,6 +50,7 @@ class AugmentedDataset(VisionDataset):
for idx, img in enumerate(self.sup_data):
self.sup_data[idx]= Image.fromarray(img) #to PIL Image
self.unsup_ratio=5 #Batch size unsup = train batch size * unsup_ratio
self.unsup_data=[]
self.unsup_targets=[]