Skip to content

Instantly share code, notes, and snippets.

@MathieuTuli
MathieuTuli / standalone_cfm_mnist.py
Created January 12, 2025 19:07
Standalone Conditional Flow Matching (CFM) for Image Generation
from functools import partial
from typing import Tuple
from pathlib import Path
from torchvision import datasets, transforms
import matplotlib.pyplot as plt
import torch.nn as nn
import torchvision
import torch