a demo of using Conditional Flow Matching to generate mnist data

安装

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

pip install torchcfm

python download_mnist.py

python train.py

python inference.py

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
checkpoints		checkpoints
LICENSE		LICENSE
README.md		README.md
download_mnist.py		download_mnist.py
inference.py		inference.py
test.py		test.py
train.py		train.py