config.json

{
  "hiddens": [
    100,
    100,
    100,
    100
  ],
  "dim": 2,
  "std": 1.0,
  "scale": 10.0,
  "num_epochs": 20000,
  "batch_size": 2000,
  "learning_rate": 0.001,
  "eta": 0.1,
  "model_type": "flow_matching_mlp",
  "description": "Flow Matching model with MLP vector field",
  "license": "mit",
  "tags": [
    "diffusion",
    "flow-matching",
    "mlp",
    "gaussian-mixture"
  ]
}