clearState.lua

-- Usage:
-- find . -name model.net -print0 | xargs -0 -n 1 th ~/clearState.lua

require 'cudnn'

local name = arg[1]

assert(paths.filep(name))

print'before'
os.execute('ls -alh '..name)

net = torch.load(name):clearState()
torch.save(name, net)

print'after'
os.execute('ls -alh '..name)