Source code for cogdl.models.base_model

from typing import Optional, Type, Any

import torch.nn as nn

from cogdl.trainers.base_trainer import BaseTrainer

[docs]class BaseModel(nn.Module): @staticmethod
[docs] def add_args(parser): """Add model-specific arguments to the parser.""" pass
[docs] def build_model_from_args(cls, args): """Build a new model instance.""" raise NotImplementedError( "Models must implement the build_model_from_args method"
[docs] def _forward_unimplemented(self, *input: Any) -> None: # abc warning pass
[docs] def get_trainer(taskType: Any, args: Any) -> Optional[Type[BaseTrainer]]: return None