parameters trained   test acc, epoch 10     50       300
--------------------------------------------------------------------------------
nn.ModuleList([...])        9219 / 9219              55.5%        84.9%    84.4%
plain list [...]             195 / 9219              36.2%        53.6%    61.3%