Demonstration of in-browser neural network inference on Kim CNN in the browser. See an online demo here.
Active code refactoring in-progress (Aug 15, 2018)
There is no setup required to run this repo out-of-the-box. However, you can train your own Kim CNN model with the following instructions.
You can train the Kim CNN model and export the trained model as an ONNX protobuf after cloning and setting up Castor. Our code is based on PyTorch v0.4 and ONNX release 1.2.2.
For example, you can do this by running the following in the Castor repo:
python -m kim_cnn --dataset SST-1 --mode static --lr 0.3213 --weight_decay 0.0002 --dropout 0.4 --onnx
Copy the model (kimcnn_static.onnx) into the onnx subdirectory.
Install protobuf for Node.js: npm install protobufjs.
Run node convert_onnx_to_js.js to output a model_parameters.js file containing all the model parameters in the parameters directory.
Licensed under the Apache License, Version 2.0.