ai-toolkit / jobs /process /TrainFineTuneProcess.py
rahul7star's picture
boilerplate
fcc02a2 verified
from collections import OrderedDict
from jobs import TrainJob
from jobs.process import BaseTrainProcess
class TrainFineTuneProcess(BaseTrainProcess):
def __init__(self,process_id: int, job: TrainJob, config: OrderedDict):
super().__init__(process_id, job, config)
def run(self):
# implement in child class
# be sure to call super().run() first
pass