We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ea0b101 commit 7e698efCopy full SHA for 7e698ef
init.lua
@@ -319,6 +319,12 @@ local function removeGradParams(net, opts)
319
m[k]:set()
320
end
321
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
328
end)
329
net.__gradParamsInfo = storages
330
@@ -342,6 +348,8 @@ local function addGradParams(net)
342
348
m[v]:set(storage, tOffset, tSize, tStride)
343
349
344
350
351
+ -- add back original getParameters
352
+ m.getParameters = nil
345
353
346
354
net.__gradParamsInfo = nil
347
355
0 commit comments