MNIST_Pytorch_python_and_capi

Neural Network Training Demo

An example project demonstrating how to train and deploy a neural network in Python and C++ using PyTorch 1.0

This is an example of how to train a MNIST network in Python and run it in c++ with pytorch 1.0

GitHub

96 stars
2 watching
8 forks
Language: Python
last commit: about 6 years ago

Related projects:

Repository Description Stars
neuralhydrology/neuralhydrology A Python library for training neural networks with focus on hydrological applications using PyTorch. 364
kimhc6028/forward-thinking-pytorch An implementation of a novel neural network training method that builds and trains networks one layer at a time. 65
prabhuomkar/pytorch-cpp A C++ implementation of PyTorch tutorials 1,965
graal-research/poutyne A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities 569
koz4k/dni-pytorch Decoupled Neural Interfaces using Synthetic Gradients for PyTorch 236
erotemic/netharn A PyTorch framework for managing and automating deep learning training loops with features like hyperparameter tracking and single-file deployments. 39
adambielski/capsnet-pytorch An implementation of a deep learning model for image classification using capsule networks with dynamic routing. 489
pemami4911/neural-combinatorial-rl-pytorch An implementation of Neural Combinatorial Optimization with Reinforcement Learning using PyTorch. 558
eladhoffer/convnet.pytorch A PyTorch implementation of various deep convolutional networks for efficient training and evaluation on diverse datasets. 347
benedekrozemberczki/capsgnn A PyTorch implementation of a graph neural network architecture 1,246
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
open-mmlab/mmengine Provides a flexible and configurable framework for training deep learning models with PyTorch. 1,179
bamos/densenet.pytorch An implementation of a deep neural network architecture in PyTorch 832
chenxi116/pnasnet.pytorch PyTorch implementation of PNASNet-5 architecture 317
pytorch/tnt A collection of tools and utilities for building and training neural networks with PyTorch. 1,663