lcnn/models/base/base_trainer.py

from abc import ABC, abstractmethod


class BaseTrainer(ABC):
    def __init__(self,
                 model=None,
                 dataset=None,
                 device='cuda',
                 **kwargs):
        #
        self.model = model
        self.dataset = dataset
        self.device=device

    # @abstractmethod
    # def train_cfg(self,model,cfg):
    #     return

    # @abstractmethod
    # def train(self,model, **kwargs):
    #     return