You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by cm...@apache.org on 2018/11/16 16:36:15 UTC
[incubator-mxnet] 02/02: do for n
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
commit e0452776d8b1113ca95b65932293b2b327ee0eaf
Author: gigasquid <cm...@gigasquidsoftware.com>
AuthorDate: Fri Nov 16 11:34:12 2018 -0500
do for n
---
.../examples/gan/src/gan/gan_mnist.clj | 33 ++++++++++++----------
1 file changed, 18 insertions(+), 15 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 c1e635e..f95ea10 100644
--- a/contrib/clojure-package/examples/gan/src/gan/gan_mnist.clj
+++ b/contrib/clojure-package/examples/gan/src/gan/gan_mnist.clj
@@ -362,32 +362,35 @@
(train devs)))
-(defn explore-flans [epoch]
+(defn explore-flans [epoch num]
(let [mod-g (-> (m/load-checkpoint {:contexts [(context/default-context)]
- :data-names ["rand"]
+ :data-names ["rand"]
:label-names [""]
:prefix "model-g"
:epoch epoch
:load-optimizer-states true})
(m/bind {:data-shapes (mx-io/provide-data rand-noise-iter)})
(m/init-params {:initializer (init/normal 0.02)})
- (m/init-optimizer {:optimizer (opt/adam {:learning-rate lr :wd 0.0 :beta1 beta1})}))
- rbatch (mx-io/next rand-noise-iter)
- out-g (-> mod-g
- (m/forward rbatch)
- (m/outputs))]
- (println "Generating image from " epoch)
- (viz/im-sav {:title (str "explore-" epoch "-" 0)
- :output-path output-path
- :x (ffirst out-g)
- :flip false})
- (-> (img/load-image (str "results/explore-" epoch "-" 0 ".jpg"))
- (img/show))))
+ (m/init-optimizer {:optimizer (opt/adam {:learning-rate lr :wd 0.0 :beta1 beta1})}))]
+
+ (println "Generating images from " epoch)
+ (dotimes [i num]
+ (let [rbatch (mx-io/next rand-noise-iter)
+ out-g (-> mod-g
+ (m/forward rbatch)
+ (m/outputs))]
+ (viz/im-sav {:title (str "explore-" epoch "-" i)
+ :output-path output-path
+ :x (ffirst out-g)
+ :flip false})
+ (-> (img/load-image (str "results/explore-" epoch "-" i ".jpg"))
+ (img/show))))))
(comment
(train [(context/cpu)])
- (explore-flans 192)
+
+ (explore-flans 195 2)
)