How to fine-tune custom datsaet with pretrained model?

gnobitab / RectifiedFlow

Official Implementation of Rectified Flow (ICLR2023 Spotlight)

964 stars 58 forks source link

How to fine-tune custom datsaet with pretrained model? #23

Open saltandlight opened 5 months ago

saltandlight commented 5 months ago

I tried to fine-tune custom dataset with pretrained model. But dataset_stats.npz file like assets/stats/celeba_stats.npz is needed. No explanation of generating dataset_stats.npz .... So How can I try to fine-tune custom datset with pretrained model?