ai-toolkit/jobs
Plat 79b4e04b80
Feat: Wandb logging (#95)
* wandb logging

* fix: start logging before train loop

* chore: add wandb dir to gitignore

* fix: wrap wandb functions

* fix: forget to send last samples

* chore: use valid type

* chore: use None when not type-checking

* chore: resolved complicated logic

* fix: follow log_every

---------

Co-authored-by: Plat <github@p1at.dev>
Co-authored-by: Jaret Burkett <jaretburkett@gmail.com>
2024-09-19 20:01:01 -06:00
..
process Feat: Wandb logging (#95) 2024-09-19 20:01:01 -06:00
BaseJob.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
ExtensionJob.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
ExtractJob.py Adjusted some names and allow to ignore text encoder on extract 2023-07-16 12:52:39 -06:00
GenerateJob.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
MergeJob.py Setup the base process for merging things. WIP 2023-07-20 07:39:31 -06:00
ModJob.py Added Model rescale and prepared a release upgrade 2023-08-01 13:49:54 -06:00
TrainJob.py Huge memory optimizations, many big fixes 2023-08-27 17:48:02 -06:00
__init__.py Added extensions and an example extension that merges models 2023-08-04 09:37:24 -06:00