Skip to content

Customized convolutional neural network in Pytorch.

Notifications You must be signed in to change notification settings

chocoluffy/pytorch_cnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 

Repository files navigation

Customized convolutional neural neworks trained with CIFAR-10 dataset.

Architecture:

  • Use 64 11*11 filters for the first convolutional layer.
  • followed by 2*2 max pooling layer (stride of 2).
  • The next two convolutional layers use 128 3*3 filters followed by the ReLU activation function.
  • batch normalization layers added right after each convolutional layers.

About

Customized convolutional neural network in Pytorch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages