Big fixes and added method to standardize values in both latent and pixel space before feeding into the network. Target values were determined over huge generated regularization sets.

This commit is contained in:
Jaret Burkett
2023-12-26 06:19:48 -07:00
parent 27ad79053e
commit d11ed7f66c
3 changed files with 64 additions and 2 deletions

View File

@@ -257,6 +257,10 @@ class TrainConfig:
if match_adapter_assist and self.match_adapter_chance == 0.0:
self.match_adapter_chance = 1.0
# standardize inputs to the meand std of the model knowledge
self.standardize_images = kwargs.get('standardize_images', False)
self.standardize_latents = kwargs.get('standardize_latents', False)
class ModelConfig:
def __init__(self, **kwargs):