config.json

{
  "channels": [
    32,
    64,
    128
  ],
  "num_residual_layers": 2,
  "t_embed_dim": 40,
  "y_embed_dim": 40,
  "num_epochs": 5000,
  "batch_size": 250,
  "learning_rate": 0.001,
  "eta": 0.1,
  "p_simple_shape": [
    1,
    32,
    32
  ],
  "num_classes": 11,
  "model_type": "flow_matching_mnist",
  "description": "Flow Matching model trained on MNIST with CFG",
  "license": "mit",
  "tags": [
    "diffusion",
    "flow-matching",
    "mnist",
    "conditional-generation"
  ]
}