michaelgzhang 303c3410e2 expand details in tensorboard logs ...

- Update tensorboard logging to track both unet and textencoder learning rates
- Update tensorboard logging to track both current and moving average epoch loss
- Clean up tensorboard log variable names for dashboard formatting

2023-01-18 13:10:13 -06:00

..

__init__.py

Proposed file structure rework and required file changes

2022-12-18 15:24:46 -05:00

model_util.py

move convert_vae to inline, restore comments

2023-01-14 21:24:09 +09:00

train_util.py

expand details in tensorboard logs

2023-01-18 13:10:13 -06:00