search

CarperAI / DRLX

Diffusion Reinforcement Learning Library
MIT License
171 stars 7 forks source link

How to see sampled images? #15

Closed TingTingin closed 1 year ago

TingTingin commented 1 year ago

How to see the sampled images that are being used for the reinforcement while training?

shahbuland commented 1 year ago

Do you want to be able to see all the sampled images each epoch? Currently it only shows one batch. What would the desirable behavior be here; logging them to wandb?

TingTingin commented 1 year ago

I want to want to be able to see all of them but I'm unsure where they are saved

shahbuland commented 1 year ago

They aren't saved... what would be the motivation behind wanting them saved if you don't mind my asking? Should be an easy enough feature to add though.

TingTingin commented 1 year ago

So you can see how the training is developing and how the model is progressing

tmabraham commented 1 year ago

@TingTingin the Weights and Biases tracking shows this and logs image generations after every epoch.

shahbuland commented 1 year ago

Solved in #20