Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

update config

Browse Source
This commit is contained in:
Vladimir Mandic
2023-01-19 23:14:40 -05:00
parent ae08ea8665
commit 6715da78bd
3 changed files with 38 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cli/train.py
+5 -3
View File
@@ -146,7 +146,7 @@ async def preprocess(params):
log.error({ 'preprocess video extract': 'no images' })
elif os.path.isdir(params.src):
if params.preprocess == 'builtin':
res = preprocess_builtin(params)
res = await preprocess_builtin(params)
if params.preprocess == 'custom':
t0 = time.perf_counter()
args.preprocess.process_src = params.src
@@ -273,12 +273,14 @@ async def train(params):
if params.grad == -1:
args.train_embedding.gradient_step = len(imgs) // args.train_embedding.batch_size
divisor = args.train_embedding.gradient_step // 60
args.train_embedding.gradient_step = args.train_embedding.gradient_step // (1 + divisor)
log.info({ 'dynamic gradient step': args.train_embedding.gradient_step })
if params.steps == -1:
args.train_embedding.steps = params.maxsteps // args.train_embedding.gradient_step
log.info({ 'dynamic steps': args.train_embedding.steps })
epoch_size = args.train_embedding.batch_size * args.train_embedding.gradient_step
epoch_size = args.train_embedding.batch_size * args.train_embedding.gradient_step
if args.train_embedding.create_image_every == -1:
args.train_embedding.create_image_every = args.train_embedding.steps // 10
if args.train_embedding.save_embedding_every == -1:
config.json
+1 -1
View File
@@ -60,7 +60,7 @@
"training_enable_tensorboard": true,
"training_tensorboard_save_images": false,
"training_tensorboard_flush_every": 120,
"sd_model_checkpoint": "sd-v15-runwayml.ckpt",
"sd_model_checkpoint": "sd-v15-runwayml.ckpt [cc6cb27103]",
"sd_checkpoint_cache": 0,
"sd_vae_checkpoint_cache": 0,
"sd_vae": "vae-ft-mse-840000-ema-pruned.ckpt",
ui-config.json
+32 -1
View File
@@ -734,5 +734,36 @@
"txt2img/Styles/visible": false,
"txt2img/Styles/value": [],
"img2img/Styles/visible": false,
"img2img/Styles/value": []
"img2img/Styles/value": [],
"modelmerger/Bake in VAE/visible": true,
"modelmerger/Bake in VAE/value": "None",
"train/Auto-sized crop/visible": true,
"train/Auto-sized crop/value": false,
"train/Dimension lower bound/visible": true,
"train/Dimension lower bound/value": 384,
"train/Dimension lower bound/minimum": 64,
"train/Dimension lower bound/maximum": 2048,
"train/Dimension lower bound/step": 8,
"train/Dimension upper bound/visible": true,
"train/Dimension upper bound/value": 768,
"train/Dimension upper bound/minimum": 64,
"train/Dimension upper bound/maximum": 2048,
"train/Dimension upper bound/step": 8,
"train/Area lower bound/visible": true,
"train/Area lower bound/value": 4096,
"train/Area lower bound/minimum": 4096,
"train/Area lower bound/maximum": 4194304,
"train/Area lower bound/step": 1,
"train/Area upper bound/visible": true,
"train/Area upper bound/value": 409600,
"train/Area upper bound/minimum": 4096,
"train/Area upper bound/maximum": 4194304,
"train/Area upper bound/step": 1,
"train/Resizing objective/visible": true,
"train/Resizing objective/value": "Maximize area",
"train/Error threshold/visible": true,
"train/Error threshold/value": 0.1,
"train/Error threshold/minimum": 0,
"train/Error threshold/maximum": 1,
"train/Error threshold/step": 0.01
}