This is an automated email from the ASF dual-hosted git repository.
cmeier pushed a commit to branch can-you-gan
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/can-you-gan by this push:
new beeada6 tweaks
beeada6 is described below
commit beeada6a015f1085327b6eba5671cfb9369f28a1
Author: gigasquid <[email protected]>
AuthorDate: Wed Nov 14 18:37:02 2018 -0500
tweaks
---
.../clojure-package/examples/gan/src/gan/gan_mnist.clj | 17 +++++++++--------
1 file changed, 9 insertions(+), 8 deletions(-)
diff --git a/contrib/clojure-package/examples/gan/src/gan/gan_mnist.clj
b/contrib/clojure-package/examples/gan/src/gan/gan_mnist.clj
index ac2293c..ea67800 100644
--- a/contrib/clojure-package/examples/gan/src/gan/gan_mnist.clj
+++ b/contrib/clojure-package/examples/gan/src/gan/gan_mnist.clj
@@ -40,22 +40,22 @@
(def data-dir "data/")
(def output-path "results/")
(def batch-size 5)
-(def num-epoch 30)
+(def num-epoch 10)
(io/make-parents (str output-path "gout"))
(defn last-saved-model-number []
- (some->> "."
+ (some->> "model"
clojure.java.io/file
file-seq
(filter #(.isFile %))
(map #(.getName %))
(filter #(clojure.string/includes? % "model-d"))
- reverse
- first
- (re-seq #"\d{4}")
- first
- Integer/parseInt))
+ (map #(re-seq #"\d{4}" %))
+ (map first)
+ (map #(when % (Integer/parseInt %)))
+ (sort)
+ (last)))
@@ -70,7 +70,8 @@
(def flan-iter (mx-io/image-record-iter {:path-imgrec "flan-128.rec"
:data-shape [3 128 128]
- :batch-size batch-size}))
+ :batch-size batch-size
+ :shuffle true}))
(defn normalize-rgb [x]
(/ (- x 128.0) 128.0))
