(self, sequence_model, structure_model)
| 76 | class FusionNetwork(nn.Module, core.Configurable): |
| 77 | |
| 78 | def __init__(self, sequence_model, structure_model): |
| 79 | super(FusionNetwork, self).__init__() |
| 80 | self.sequence_model = sequence_model |
| 81 | self.structure_model = structure_model |
| 82 | self.output_dim = structure_model.output_dim |
| 83 | |
| 84 | def forward(self, graph, input, all_loss=None, metric=None, batch=None): |
| 85 | # Sequence model |
nothing calls this directly
no outgoing calls
no test coverage detected