pytorch-CNN

This is a simple code implementing popular paper about machine-learning. (Using python, pytorch, single GPU)

To simplify the code, it may be different from the original paper.

Basic model

• MLP - MNIST dataset
• CNN - CIFAR10 dataset

Model	Dataset	Description
MLP	MNIST	Multi Layer Perceptron
CNN	CIFAR 10	Convolution Neural Network

Computer Vision paper implement

• AlexNet - CIFAR10 dataset
• ResNet - CIFAR10 dataset
• VGGNet - CIFAR10 dataset
• GAN - MNIST dataset

Model	Dataset	Description	paper
AlexNet	CIFAR 10	ImageNet Classification with Deep Convolutional Neural Networks(2012)	AlexNet
ResNet	CIFAR 10	Deep Residual Learning for Image Recognition(2016)	ResNet
VGGNet	CIFAR 10	VERY DEEP CONVOLUTIONAL NETWORKS FOR LARGE-SCALE IMAGE RECOGNITION(2014)	VGGNet
GAN	MNIST	Generative Adversarial Nets	GAN
Faster R-CNN	-----	Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks	Faster R-CNN

NLP paper implement

• Attention - ChatbotData.csv

Model	Dataset	Description	paper
Transformer Model - using Attention	ChatbotData.csv	Attention is all you need Review(2017)	Attention