Skip to content

vabh/convolutional-neural-fabrics

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch implementation of Convolutional Neural Fabrics arxiv:1606.02492 There are some minor differences:

  • The raw image is first convolved, to obtain #channels feature maps.
  • The upsampling is followed by a convolution, and the result is then summed with the other inputs. In the paper, they first sum and then convolve on the result.
  • These can be easily changed in the UpSample, DownSample, SameRes class definitions inside neural_fabrics.py. Feel free to implement your own procedure and experiment.

To run on CIFAR-10:

python neural_fabric.py --dataset cifar10 --save fabric_cifar10

Test set error: 7.2%, with rotation and translation augmented training data.

About

PyTorch implementation of Convolutional Neural Fabrics http://arxiv.org/abs/1606.02492

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages