Skip to content

suggestions #318

@Hugs288

Description

@Hugs288
  1. add validation loss to the ui (Validation loss  kohya-ss/sd-scripts#1864) (currently doesnt work at all because extra args converts everything every arg to string)

  2. combine sdpa/xformers, caching/to disk options to a single dropdown selector

  3. on networks tab:
    dora can be its own network type
    wd on output should be enabled by default for dora and can be removed from ui
    bypass mode can be removed from ui (people who know they need it can use it via network args)
    cache te outputs seems useless to me, why not just train unet only?

  4. improve default settings:
    gradiant checkpointing on by default
    if sdxl is checked set resolution to 1024 and max bucket res to 2048
    change default lr to 3e-4, te lr to 3e-5
    remove weight_decay=0.1 optional optimizer arg
    network dim/alpha = 16/16 or 16/8
    lr scheduler either rex with 1 cycle or constant with warmup
    10% warmup by default

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions