Skip to content

Commit

Permalink
Merge pull request #11 from tianyizheng02/master
Browse files Browse the repository at this point in the history
super already calls forward_graph connection
  • Loading branch information
Vivswan authored Jan 13, 2023
2 parents 2f86972 + 1da1e7b commit 9e1a6f4
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion analogvnn/nn/module/FullSequential.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,5 @@ def compile(self, device: Optional[torch.device] = None, layer_data: bool = True
FullSequential: self
"""
arr = [self.graphs.INPUT, *list(self._runtime_module_list.values()), self.graphs.OUTPUT]
self.graphs.forward_graph.add_connection(*arr)
self.graphs.backward_graph.add_connection(*reversed(arr))
return super().compile(device, layer_data)

0 comments on commit 9e1a6f4

Please sign in to comment.