Skip to content

Instantly share code, notes, and snippets.

@farrajota
farrajota / freeze.lua
Created September 6, 2016 14:17
freeze parameters of a layer
model.modules[1].parameters = function() return nil end -- freezes the layer when using optim
model.modules[1].accGradParameters = function() end -- overwrite this to reduce computations