xieyanfu / fastnet

A convolutional neural network framework.

Fastnet

A convolutional neural network framework, building on top of the convolutional kernel code from cuda-convnet.

Setup

git clone https://github.com/rjpower/fastnet
cd fastnet
python setup.py develop [--user]

Usage

To run a trainer directly:

python fastnet/trainer.py --help

Take a look at the scripts in fastnet/scripts for examples of how to run your own network.

Requires

• NumPy
• CUDA
• PyCUDA

Referencing

If you use FastNet for your research, please cite us in your publication:

R. Power and J. Lin.
FastNet -- A framework for convolutional neural networks.
https://github.com/rjpower/fastnet

Bibtex:

@misc{PowerFastNet,
      author = {Power, Russell and Lin, Justin},
      title = {{F}ast{N}et -- {A} framework for convolutional neural networks},
      howpublished = {\url{https://github.com/rjpower/fastnet}},
}