ashleve / lightning-hydra-template

PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡
4.13k stars 643 forks source link

How to save images in testing_setp ? #631

Closed YChienHung closed 8 months ago

YChienHung commented 8 months ago

I will generate some images in testing_step, and they will be saved in a path.

Can you give me some ideas?