Skip to content

Commit

Permalink
drcn fixed
Browse files Browse the repository at this point in the history
  • Loading branch information
ghif committed Feb 28, 2016
1 parent c807f6a commit 26b3253
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions models.py
Original file line number Diff line number Diff line change
Expand Up @@ -407,15 +407,15 @@ def __init__(self, input_shape, output_dim, net_config, ae_config, enc_weights=N
self.net_config = net_config
self.ae_config = ae_config

self.enc_weights = weights
self.enc_weights = enc_weights

self.create_architecture(input_shape, dense_dim=net_config['dense_dim'],
dout=output_dim, dropout=net_config['dropout'])
self.compile(lr_net=net_config['lr'], lr_ae=ae_config['lr'])

def create_architecture(self, input_shape, dense_dim=1024, dout=10, dropout=0.5):
self.convnet_.create_architecture(input_shape, dout=dout, dropout=dropout,
input_var_=self.X_, output_var_=self.Yout_)
input_var_=self.X_, output_var_=self.Yout_, enc_weights=self.enc_weights)

self.convae_.create_architecture(input_shape, input_var_=self.X_,
output_var_=self.Xout_, convnet_=self.convnet_)
Expand Down

0 comments on commit 26b3253

Please sign in to comment.