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"
)