MCPcopy Index your code
hub / github.com/deepspeedai/DeepSpeedExamples / set_device

Method set_device

bing_bert/turing/models.py:114–115  ·  view source on GitHub ↗
(self, device)

Source from the content-addressed store, hash-verified

112 self.device = None
113
114 def set_device(self, device):
115 self.device = device
116
117 def save(self, filename: str):
118 network=self.network.module

Callers 7

prepare_model_optimizerFunction · 0.95
mainFunction · 0.80
mainFunction · 0.80
initialize_distributedFunction · 0.80
initialize_distributedFunction · 0.80
initialize_distributedFunction · 0.80
initialize_distributedFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected