Model is None
shi27feng opened this issue · comments
Hi @vballoli,
It seems to have a bug in the code of AGC:
if model is not None:
assert ignore_agc not in [None, []], "You must specify ignore_agc for AGC to ignore fc-like(or other) layers"
names = [name for name, module in model.named_modules()]
for module_name in ignore_agc:
if module_name not in names:
raise ModuleNotFoundError("Module name {} not found in the model".format(module_name))
params = [{"params": list(module.parameters())} for name,
module in model.named_modules() if name not in ignore_agc]
else:
params = [{"params": list(module.parameters())} for name,
module in model.named_modules()]
When model is None
then the else
part of the code cannot get name
and module
from model.named_modules()
.
Thanks
My bad, thanks for raising this issue. I'll add a fix soon.
Fixed in the latest commit. Thanks for reporting!