Source code for models.base_model

import torch
import torch.nn as nn


[docs]class BaseModel(nn.Module): @staticmethod
[docs] def add_args(parser): """Add model-specific arguments to the parser.""" pass
@classmethod
[docs] def build_model_from_args(cls, args): """Build a new model instance.""" raise NotImplementedError( "Models must implement the build_model_from_args method"
)