MCPcopy Create free account
hub / github.com/pytorch/tutorials / forward

Method forward

advanced_source/rpc_ddp_tutorial/main.py:33–35  ·  view source on GitHub ↗
(self, indices, offsets)

Source from the content-addressed store, hash-verified

31 self.device = device
32
33 def forward(self, indices, offsets):
34 emb_lookup = self.remote_emb_module.forward(indices, offsets)
35 return self.fc(emb_lookup.cuda(self.device))
36# END hybrid_model
37
38# BEGIN setup_trainer

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected