Setup base for training jobs. Added sd-scripts as a submodule
This commit is contained in:
25
jobs/process/BaseTrainProcess.py
Normal file
25
jobs/process/BaseTrainProcess.py
Normal 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
|
||||
@@ -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')
|
||||
13
jobs/process/TrainFineTuneProcess.py
Normal file
13
jobs/process/TrainFineTuneProcess.py
Normal 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
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user