(self, remote_emb_module, device)
| 25 | """ |
| 26 | |
| 27 | def __init__(self, remote_emb_module, device): |
| 28 | super(HybridModel, self).__init__() |
| 29 | self.remote_emb_module = remote_emb_module |
| 30 | self.fc = DDP(torch.nn.Linear(16, 8).cuda(device), device_ids=[device]) |
| 31 | self.device = device |
| 32 | |
| 33 | def forward(self, indices, offsets): |
| 34 | emb_lookup = self.remote_emb_module.forward(indices, offsets) |
nothing calls this directly
no outgoing calls
no test coverage detected