Closed toontran closed 1 year ago
Thanks for pointing this out and the fix! I will merge it in 2 weeks.
Thomas
Actually I think this is the correct behavior, since [nn.Linear(100, 100)] * 10
will stack the same layer (same object instance) 10 times, sharing the same parameters, as opposed to e.g. [nn.Linear(100, 100) for i in range(10)]
which will instantiate 10 different layers, with their own parameters.
However the current implementation cannot handle sharing parameters between layers. For instance, recurrent networks won't work.
Output (only has 3 layers in
lc
):