models.base_model

Module Contents

Classes

BaseModel

class models.base_model.BaseModel

Bases: torch.nn.Module

static add_args(parser)

Add model-specific arguments to the parser.

abstract classmethod build_model_from_args(cls, args)

Build a new model instance.