ModuleList will work too. You can try to visualise the graph to check if it’s what you want to build: https://discuss.pytorch.org/t/print-autograd-graph/692.
1 Like
ModuleList will work too. You can try to visualise the graph to check if it’s what you want to build: https://discuss.pytorch.org/t/print-autograd-graph/692.