MCPcopy Create free account
hub / github.com/pytorch/pytorch / GetDevices

Method GetDevices

caffe2/python/model_helper.py:419–422  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

417 self, blob_out, batch_size, db, db_type, **kwargs)
418
419 def GetDevices(self):
420 assert len(self._devices) > 0, \
421 "Use data_parallel_model to run model on multiple GPUs."
422 return self._devices
423
424 def __getattr__(self, op_type):
425 """Catch-all for all other operators, mostly those without params."""

Callers 1

FinalizeAfterCheckpointFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected