Thanks a lot for the clear reply!!
Should I just fix the normal dictionary like
self.classifier = nn.ModuleDict({})

then add elements??
And what do you mean by ‘not properly registered inside the module’??
Does that mean that the model won’t be able to properly record operations and data flows
for that variable??