deepmd.pt.entrypoints package

Submodules

deepmd.pt.entrypoints.main module

class deepmd.pt.entrypoints.main.SummaryPrinter

Bases: SummaryPrinter

Summary printer for PyTorch.

Methods

__call__()	Print build and current running cluster configuration summary.
get_backend_info()	Get backend information.
get_compute_device()	Get Compute device.
get_ngpus()	Get the number of GPUs.
is_built_with_cuda()	Check if the backend is built with CUDA.
is_built_with_rocm()	Check if the backend is built with ROCm.

get_backend_info() → dict

Get backend information.

get_compute_device() → str

Get Compute device.

get_ngpus() → int

Get the number of GPUs.

is_built_with_cuda() → bool

Check if the backend is built with CUDA.

is_built_with_rocm() → bool

Check if the backend is built with ROCm.

deepmd.pt.entrypoints.main.freeze(FLAGS)

deepmd.pt.entrypoints.main.get_trainer(config, init_model=None, restart_model=None, finetune_model=None, model_branch='', force_load=False, init_frz_model=None, shared_links=None)

deepmd.pt.entrypoints.main.main(args: Optional[Union[List[str], Namespace]] = None)

deepmd.pt.entrypoints.main.train(FLAGS)