Merge pull request #12 from iwatobipen/master

change loss function
cxhernandez · Jan 11, 2018 · 1d7e208 · 1d7e208
2 parents cdeb5c3 + 86523d3
commit 1d7e208
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/molencoder/utils.py b/molencoder/utils.py
@@ -127,7 +127,7 @@ def train_model(train_loader, encoder, decoder, optimizer, dtype,
         y_var = encoder(x_var)
         z_var = decoder(y_var)
 
-        loss = encoder.vae_loss(x_var, z_var)
+        loss = encoder.vae_loss(z_var, x_var)
         if (t + 1) % print_every == 0:
             print('t = %d, loss = %.4f' % (t + 1, loss.data[0]))