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