Added logging of each loss separately and image logging to W&B

Nota-NetsPresso / BK-SDM

A Compressed Stable Diffusion for Efficient Text-to-Image Generation [ECCV'24]

Other

238 stars 16 forks source link

Added logging of each loss separately and image logging to W&B #39

Closed Godofnothing closed 10 months ago

Godofnothing commented 11 months ago

Hi, sorry for delay. In accordance with your suggestions I've split W&B logging and batched generation into 2 separate commits having taken into account your comments.

bokyeong1015 commented 10 months ago

Hi, thanks again for your contribution. This PR has been merged into our main branch :)