Skip to content

Commit 7e698ef

Browse files
committed
Disable getParameters when removeGradParams=true
1 parent ea0b101 commit 7e698ef

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

init.lua

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -319,6 +319,12 @@ local function removeGradParams(net, opts)
319319
m[k]:set()
320320
end
321321
end
322+
-- disabling getParameters
323+
m.getParameters = function(self)
324+
error('getParameters was disabled by optnet '..
325+
'(by option removeGradParams=true). '..
326+
'Call optnet.removeOptimization(net) to enable it back.')
327+
end
322328
end)
323329
net.__gradParamsInfo = storages
324330
end
@@ -342,6 +348,8 @@ local function addGradParams(net)
342348
m[v]:set(storage, tOffset, tSize, tStride)
343349
end
344350
end
351+
-- add back original getParameters
352+
m.getParameters = nil
345353
end)
346354
net.__gradParamsInfo = nil
347355
end

0 commit comments

Comments
 (0)