Closed bokyeong1015 closed 10 months ago
To incorporate the below feature
In addition, the base training script src/kd_train_text_to_image.py logs only the total loss to W&B and one may be interested in each particular contribution. I added image logging to W&B as well.
To incorporate the below feature