search

bghira / SimpleTuner

A general fine-tuning kit geared toward diffusion models.
GNU Affero General Public License v3.0
1.81k stars 172 forks source link

tensorboard support #497

Closed Laidawang closed 5 months ago

Laidawang commented 5 months ago

Can you add support for tensorboard?

twri commented 5 months ago

It's already in the trainer.

--report_to REPORT_TO The integration to report the results and logs to. Supported platforms are"tensorboard" (default),"wandb"and"comet_ml". Use"all"to report to all integrations.

bghira commented 5 months ago

if it's not working, i'm not quite sure what to do about it. i've only ever used wandb.

twri commented 5 months ago

I tried it yesterday and it works but I don't use the wrapper bash scripts.

bghira commented 5 months ago

i mean, does it receive validation images too? that's the part i think isn't implemented.

Laidawang commented 5 months ago

I think it dont receive validation images，check the diffusers implementation https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/train_dreambooth_lora_sd3.py#L153

sayakpaul commented 5 months ago

Yeah we don't send the validation images to Tensorboard given it's used very little. But feel free to send over a PR if you think it's nice.