Reduce epoch size for deep learning excersize to reduce memory requirements

bwall · bwall · commit 020b349737ae · 2017-08-22T19:27:41.000-07:00
diff --git a/deep_learning_examples/train_model.py b/deep_learning_examples/train_model.py
@@ -12,7 +12,7 @@
 from multiprocessing import Process as Thread, Queue
 from itertools import cycle, islice
 
-EPOCH_SIZE = 16384
+EPOCH_SIZE = 2048
 train_queue = Queue(maxsize=20)
 test_queue = Queue(maxsize=20)
 
diff --git a/deep_learning_examples/train_xor.sh b/deep_learning_examples/train_xor.sh
@@ -1,2 +1,2 @@
 #!/usr/bin/env bash
-KERAS_BACKEND="theano" THEANO_FLAGS=device=gpu,floatX=float32 python train_xor_lstm.py
+KERAS_BACKEND="theano" THEANO_FLAGS=device=gpu,floatX=float32 python train_model.py

Original file line number	Diff line number	Diff line change
`@@ -1,2 +1,2 @@`
`1`	`1`	`#!/usr/bin/env bash`
`2`		`-KERAS_BACKEND="theano" THEANO_FLAGS=device=gpu,floatX=float32 python train_xor_lstm.py`
	`2`	`+KERAS_BACKEND="theano" THEANO_FLAGS=device=gpu,floatX=float32 python train_model.py`