-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Closed
Description
Right now the parameters() function of nn.Container returns the parameters of the container children, but not the container itself. Problem comes from this line:
In python 3 this can simply be fixed with:
yield from super(Container, self).parameters(memo)
Or in python 2 or 3:
for p in super(Container, self).parameters(memo):
yield p
Metadata
Metadata
Assignees
Labels
No labels