MCPcopy Create free account
hub / github.com/modelscope/modelscope / cuda

Method cuda

modelscope/models/science/unifold/modules/frame.py:450–451  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

448 return Frame(rot_obj, ca_xyz.type(input_dtype))
449
450 def cuda(self) -> Frame:
451 return Frame(self._r.cuda(), self._t.cuda())
452
453 @property
454 def dtype(self) -> torch.dtype:

Callers 15

__init__Method · 0.45
_ssim_3dFunction · 0.45
forwardMethod · 0.45
preprocessMethod · 0.45
forwardMethod · 0.45
preprocessMethod · 0.45
format_nvdiffrast_formatMethod · 0.45
preprocessMethod · 0.45
load_modelFunction · 0.45
preprocessMethod · 0.45
preprocessMethod · 0.45
__init__Method · 0.45

Calls 1

FrameClass · 0.85

Tested by 3

test_funcFunction · 0.36
test_amp_optimizer_hookMethod · 0.36
test_apex_optimizer_hookMethod · 0.36