Probable bug: `dependency.py`:550

[MPiotr]

Could you please check dependency.py lines 549-550:

        for o in flatten_as_list(out):
            module2node = self._build_graph(o.grad_fn, gradfn2module, reused)

looks like for multiple outputs the only last module2node is saved. At least for my model (with multiple output) changing to modul2node.update(self...) resolved program failure.