moreo
/
kernel_authorship
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

bugfix

This commit is contained in:
Alejandro Moreo Fernandez 2020-07-09 22:02:48 +02:00
parent ced842a219
commit be83411e25
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/model/layers.py
View File

@ -6,7 +6,7 @@ import torch.nn.functional as F
class Phi(nn.Module): class Phi(nn.Module):
def __init__(self, cnn, ff, norm): def __init__(self, cnn, ff, norm=None):
super(Phi, self).__init__() super(Phi, self).__init__()
self.cnn = cnn self.cnn = cnn
self.ff = ff self.ff = ff