deepmd.pt.infer.inference

Contents

deepmd.pt.infer.inference#

Attributes#

Classes#

Module Contents#

deepmd.pt.infer.inference.log[source]#

class deepmd.pt.infer.inference.Tester(model_ckpt: str | torch.nn.Module, head: str | None = None)[source]#

multi_task[source]#

model_params[source]#

model[source]#

modifier = None[source]#

wrapper[source]#