Added my good ole pattern loss. God I love that thing, conv transpose pattern instantly wiped from vae

This commit is contained in:
Jaret Burkett committed 2023-07-20 15:44:16 -06:00
1 parent 982e0be7a9
commit 0761656a90
4 files changed
+56 -12

No files matched your search

+3 -3
View File
@@ -15,7 +15,7 @@ sys.path.append(REPOS_ROOT)
process_dict = {
'vae': 'TrainVAEProcess',
'finetune': 'TrainFineTuneProcess'
'slider': 'TrainSliderProcess',
}
@@ -27,8 +27,8 @@ class TrainJob(BaseJob):
self.training_folder = self.get_conf('training_folder', required=True)
self.is_v2 = self.get_conf('is_v2', False)
self.device = self.get_conf('device', 'cpu')
self.gradient_accumulation_steps = self.get_conf('gradient_accumulation_steps', 1)
self.mixed_precision = self.get_conf('mixed_precision', False) # fp16
# self.gradient_accumulation_steps = self.get_conf('gradient_accumulation_steps', 1)
# self.mixed_precision = self.get_conf('mixed_precision', False) # fp16
self.log_dir = self.get_conf('log_dir', None)
self.writer = None