Kim CNN visualization
Active code refactoring in-progress (Aug 15, 2018)
Training Your Own Kim CNN Model
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
Install protobuf for Node.js:
npm install protobufjs.
node convert_onnx_to_js.js to output a
model_parameters.js file containing all the model parameters in the
Licensed under the Apache License, Version 2.0.