cogdl.models.base_model

Module Contents

Classes

BaseModel

class cogdl.models.base_model.BaseModel[source]

Bases: torch.nn.Module

static add_args(parser)[source]

Add model-specific arguments to the parser.

abstract classmethod build_model_from_args(cls, args)[source]

Build a new model instance.

_forward_unimplemented(self, *input: Any) → None[source]
static get_trainer(taskType: Any, args: Any) → Optional[Type[BaseTrainer]][source]