acobobby:

where new_units in this case is a list of modules containing parameters.


There are no new modules. The neurons added are to existing layers, not new layers. When I print the network parameters they show up in the same parameter group.