Closed fankaisheng closed 1 year ago
https://github.com/learnables/learn2learn/blob/master/learn2learn/utils/__init__.py in line 239 function 'update_module' , it has no return value, but in line 308:
for module_key in module._modules: module._modules[module_key] = update_module( module._modules[module_key], updates=None, memo=memo, ) This will eventually cause the model defined by nn.Sequential to become none per layer
I'm not sure I understand: update_module returns the updated module on line 320, right?
update_module
https://github.com/learnables/learn2learn/blob/master/learn2learn/utils/__init__.py in line 239 function 'update_module' , it has no return value, but in line 308:
for module_key in module._modules: module._modules[module_key] = update_module( module._modules[module_key], updates=None, memo=memo, ) This will eventually cause the model defined by nn.Sequential to become none per layer