examples_changed

MelJan · Apr 24, 2017 · 6a4f6b9 · 6a4f6b9
1 parent 5a86f1d
commit 6a4f6b9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/RBM_MNIST_big.py b/examples/RBM_MNIST_big.py
@@ -92,7 +92,10 @@
     # Loop over all batches
     for b in range(0, train_data.shape[0], batch_size):
         batch = train_data[b:b + batch_size, :]
-        trainer_pcd.train(data=batch, epsilon=0.01)
+        trainer_pcd.train(data=batch,
+                          epsilon=0.01,
+                          update_visible_offsets=update_offsets,
+                          update_hidden_offsets=update_offsets)
 
     # Calculate reconstruction error and expected end time every 10th epoch
     if epoch % 10 == 0: