Skip to content

Implementation of http://arxiv.org/abs/1511.05641 that lets one build a larger net starting from a smaller one.

Notifications You must be signed in to change notification settings

dasguptar/net2net.torch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

  • Proof of concept with unit tests
  • Does not handle batchnorm cases yet
n2n = require 'net2net'

-- net  = network
-- pos1 = position at which one has to widen the output
-- pos2 = position at which the next weight layer is present
-- newWidth   = new width of the layer
n2n.wider(net, pos1, pos2, newWidth)

-- pos = position at which the layer has to be deepened
-- nonlin = type of non-linearity to insert
n2n.deeper(net, pos, nonlin)

Example usage in test.lua

About

Implementation of http://arxiv.org/abs/1511.05641 that lets one build a larger net starting from a smaller one.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Lua 97.7%
  • CMake 2.3%