Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
css
 
 
js
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Kim CNN visualization

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)

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 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.

License

Licensed under the Apache License, Version 2.0.

About

An in-browser visualization of Kim CNN

Resources

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •