Setup base for training jobs. Added sd-scripts as a submodule

This commit is contained in:
Jaret Burkett
2023-07-08 13:50:59 -06:00
parent 37354b006e
commit 47d094e528
13 changed files with 151 additions and 18 deletions

View File

@@ -0,0 +1,25 @@
from collections import OrderedDict
from jobs import TrainJob
from jobs.process.BaseProcess import BaseProcess
class BaseTrainProcess(BaseProcess):
job: TrainJob
process_id: int
config: OrderedDict
def __init__(
self,
process_id: int,
job: TrainJob,
config: OrderedDict
):
super().__init__(process_id, job, config)
self.process_id = process_id
self.job = job
self.config = config
def run(self):
# implement in child class
# be sure to call super().run() first
pass

View File

@@ -27,7 +27,7 @@ mode_dict = {
}
class LoconExtractProcess(BaseExtractProcess):
class ExtractLoconProcess(BaseExtractProcess):
def __init__(self, process_id: int, job: ExtractJob, config: OrderedDict):
super().__init__(process_id, job, config)
self.mode = self.get_conf('mode', 'fixed')

View File

@@ -0,0 +1,13 @@
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

View File

@@ -1,3 +1,5 @@
from .BaseExtractProcess import BaseExtractProcess
from .LoconExtractProcess import LoconExtractProcess
from .ExtractLoconProcess import ExtractLoconProcess
from .BaseProcess import BaseProcess
from .BaseTrainProcess import BaseTrainProcess
from .TrainFineTuneProcess import TrainFineTuneProcess