deepmd.pt.model.backbone

Submodules

Classes

Package Contents

class deepmd.pt.model.backbone.BackBone(**kwargs)[source]

Bases: torch.nn.Module

abstract forward(**kwargs)[source]: Calculate backBone.