MCPcopy
hub / github.com/TencentARC/Pixal3D / cuda

Method cuda

pixal3d/pipelines/base.py:68–69  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

66 model.to(device)
67
68 def cuda(self) -> None:
69 self.to(torch.device("cuda"))
70
71 def cpu(self) -> None:
72 self.to(torch.device("cpu"))

Callers 15

mainFunction · 0.45
init_modelsFunction · 0.45
unpack_stateFunction · 0.45
generate_3dFunction · 0.45
init_pipelineFunction · 0.45
mainFunction · 0.45
encode_shape_latent.pyFile · 0.45
encode_shape_latent_view.pyFile · 0.45
encode_ss_latent.pyFile · 0.45
load_latentFunction · 0.45
mainFunction · 0.45
encode_pbr_latent.pyFile · 0.45

Calls 2

toMethod · 0.95
deviceMethod · 0.45

Tested by

no test coverage detected