summaryrefslogtreecommitdiffstats
path: root/train_ti.py
Commit message (Collapse)AuthorAgeFilesLines
...
* Removed unused code, put training callbacks in dataclassVolpeon2023-01-151-48/+1
|
* Added functional TI strategyVolpeon2023-01-151-78/+30
|
* Added functional trainerVolpeon2023-01-151-26/+23
|
* UpdateVolpeon2023-01-141-5/+5
|
* UpdateVolpeon2023-01-141-3/+4
|
* WIP: Modularization ("free(): invalid pointer" my ass)Volpeon2023-01-141-59/+15
|
* TI: Prepare UNet with Accelerate as wellVolpeon2023-01-141-12/+15
|
* FixVolpeon2023-01-141-2/+2
|
* CleanupVolpeon2023-01-141-21/+12
|
* Unified training script structureVolpeon2023-01-131-3/+6
|
* Reverted modularization mostlyVolpeon2023-01-131-81/+386
|
* More modularizationVolpeon2023-01-131-409/+70
|
* Simplified step calculationsVolpeon2023-01-131-13/+11
|
* Removed PromptProcessor, modularized training loopVolpeon2023-01-131-215/+53
|
* Added TI decay start offsetVolpeon2023-01-131-2/+8
|
* Code deduplicationVolpeon2023-01-131-60/+26
|
* UpdateVolpeon2023-01-121-16/+14
|
* Fixed TI decayVolpeon2023-01-121-8/+3
|
* Disable Adam weight decayVolpeon2023-01-121-1/+1
|
* FixVolpeon2023-01-111-3/+3
|
* HeckVolpeon2023-01-111-1/+1
|
* TI: Use grad clipping from LoRA #104Volpeon2023-01-111-8/+11
|
* Better defaultsVolpeon2023-01-101-4/+4
|
* FixVolpeon2023-01-101-1/+1
|
* Added arg to disable tag shufflingVolpeon2023-01-101-6/+16
|
* Enable buckets for validation, fixed vaildation repeat argVolpeon2023-01-091-4/+1
|
* Add --valid_set_repeatVolpeon2023-01-091-0/+22
|
* FixVolpeon2023-01-081-1/+1
|
* Improved aspect ratio bucketingVolpeon2023-01-081-1/+15
|
* Fixed aspect ratio bucketing; allow passing token IDs to pipelineVolpeon2023-01-081-11/+13
|
* Improved aspect ratio bucketingVolpeon2023-01-081-48/+37
|
* CleanupVolpeon2023-01-071-44/+16
|
* Made aspect ratio bucketing configurableVolpeon2023-01-071-0/+13
|
* Added progressive aspect ratio bucketingVolpeon2023-01-071-48/+46
|
* UpdateVolpeon2023-01-071-3/+3
|
* Relaxed ema_power for TIVolpeon2023-01-061-1/+1
|
* UpdateVolpeon2023-01-061-24/+17
|
* Use context manager for EMA, on_train/eval hooksVolpeon2023-01-061-54/+66
|
* Add contextmanager to EMAModel to apply weights temporarilyVolpeon2023-01-061-33/+24
|
* Log EMA decayVolpeon2023-01-051-1/+2
|
* Added EMA to TIVolpeon2023-01-051-1/+58
|
* UpdateVolpeon2023-01-051-7/+7
|
* FixVolpeon2023-01-051-8/+11
|
* Various cleanupsVolpeon2023-01-051-65/+21
|
* UpdateVolpeon2023-01-041-1/+1
|
* Various updatesVolpeon2023-01-041-0/+8
|
* Better eval generatorVolpeon2023-01-041-6/+6
|
* Fixed reproducibility, more consistant validationVolpeon2023-01-041-5/+19
|
* Don't use vector_dropout by defaultVolpeon2023-01-031-1/+1
|
* Added vector dropoutVolpeon2023-01-031-5/+19
|