fine-tuning-service / src /abstract_trainer.py
fashxp's picture
initial commit
7c4332a
raw
history blame
464 Bytes
from abc import ABC, abstractmethod
import logging
from .training_status import TrainingStatus
logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)
class AbstractTrainer(ABC):
__training_status: TrainingStatus = TrainingStatus();
@abstractmethod
async def start_training(self):
logger.info('start abstract trainer training')
pass
def get_status(self) -> TrainingStatus:
return self.__training_status