Change batchsize

1080Ti can allocate only 2 batches. So changed it.
AmisArbel · Nov 10, 2018 · 86b21b6 · 86b21b6
1 parent 6a75f20
commit 86b21b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/params.py b/params.py
@@ -1,5 +1,5 @@
 # parameters of training
-batchsize = 3
+batchsize = 2
 lr = 1e-4
 trigger = (500000, 'iteration')
 annealing_interval = (200000, 'iteration')