ai-toolkit

Files

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

__init__.py

Added extensions and an example extension that merges models

2023-08-04 09:37:24 -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