baofff / U-ViT

A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to train on a new dataset?

Cassiatora opened this issue · comments

I want to train a new data set, how do I get the fid_stats?

For those who want to generate fid_stats for your dataset, I've got the stats for my dataset by running this repository with --save-stats flag.
If your dataset folder structure is nested by class or something, you can use this fork to find images recursively.