Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
add readme to customop
Browse files Browse the repository at this point in the history
  • Loading branch information
lanking520 committed Jun 26, 2018
1 parent 89b6186 commit dd39249
Showing 1 changed file with 22 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
# Custom Operator Example for Scala
This is the example using Custom Operator for type-safe api of Scala.
In the example, a `Softmax` operator is implemented to run the MNIST example.

There is also an example using RTC. However, the rtc module is depreciated and no longer can be used. Please contribute to use CudaModule operator to replace the rtc.

## Setup
### Download the source File
```$xslt
https://s3.us-east-2.amazonaws.com/mxnet-scala/scala-example-ci/mnist/mnist.zip
```
### Unzip the file
```$xslt
unzip mnist.zip
```
### Arguement Configuration
Then you need to define the arguments that you would like to pass in the model:
```$xslt
--data-path <location of your downloaded file>
```

you can find more in [here](https://github.com/apache/incubator-mxnet/blob/scala-package/examples/src/main/scala/org/apache/mxnet/examples/customop/ExampleCustomOp.scala#L218-L221)

0 comments on commit dd39249

Please sign in to comment.