Chouffe commented on a change in pull request #15340: [Clojure] Add fastText
example
URL: https://github.com/apache/incubator-mxnet/pull/15340#discussion_r297296210
##########
File path: contrib/clojure-package/examples/cnn-text-classification/README.md
##########
@@ -58,15 +72,15 @@ you'll need to unzip them and place them in the
`contrib/clojure-package/data` d
Then you can run training on a subset of examples through the repl using:
```
-(train-convnet {:embedding-size 300 :batch-size 100 :test-size 100 :num-epoch
10 :max-examples 1000 :pretrained-embedding :word2vec})
+(train-convnet {:devs [(context/cpu 0)] :embedding-size 300 :batch-size 100
:test-size 100 :num-epoch 10 :max-examples 1000 :pretrained-embedding
:word2vec})
```
Note that loading word2vec embeddings consumes memory and takes some time.
You can also train them using `JVM_OPTS="-Xmx8g" lein run` once you've modified
the parameters to `train-convnet` (see above) in
`src/cnn_text_classification/classifier.clj`.
In order to run training with word2vec on the complete data set, you will need
to run:
```
-(train-convnet {:embedding-size 300 :batch-size 100 :test-size 1000 :num-epoch
10 :pretrained-embedding :word2vec})
+(train-convnet {:devs [(context/cpu 0)] :embedding-size 300 :batch-size 100
:test-size 1000 :num-epoch 10 :pretrained-embedding :word2vec})
Review comment:
Thanks for fixing the README!
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
