deepmd.pt.entrypoints.main

Module Contents

Classes

SummaryPrinter

Summary printer for PyTorch.

Functions

get_trainer(config[, init_model, restart_model, ...])

train(FLAGS)

freeze(FLAGS)

main([args])

Attributes

log

deepmd.pt.entrypoints.main.log[source]
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)[source]
class deepmd.pt.entrypoints.main.SummaryPrinter[source]

Bases: deepmd.utils.summary.SummaryPrinter

Summary printer for PyTorch.

is_built_with_cuda() → bool[source]

Check if the backend is built with CUDA.

is_built_with_rocm() → bool[source]

Check if the backend is built with ROCm.

get_compute_device() → str[source]

Get Compute device.

get_ngpus() → int[source]

Get the number of GPUs.

get_backend_info() → dict[source]

Get backend information.

deepmd.pt.entrypoints.main.train(FLAGS)[source]
deepmd.pt.entrypoints.main.freeze(FLAGS)[source]
deepmd.pt.entrypoints.main.main(args: List[str] | argparse.Namespace | None = None)[source]