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

3
higher/model.py
View file

@ -43,7 +43,8 @@ class LeNet(nn.Module):
#print("Shape ", out.shape)
out = F.linear(out, self._params["w4"], self._params["b4"])
#print("Shape ", out.shape)
return F.log_softmax(out, dim=1)
#return F.log_softmax(out, dim=1)
return out
def __getitem__(self, key):
return self._params[key]