Upload model README
Browse files
README.md
CHANGED
|
@@ -14,7 +14,7 @@ Flow Matching model trained on MNIST with CFG
|
|
| 14 |
## Model Details
|
| 15 |
|
| 16 |
- **Model Type**: flow_matching_mnist
|
| 17 |
-
- **Training Epochs**:
|
| 18 |
- **Batch Size**: 250
|
| 19 |
- **Learning Rate**: 0.001
|
| 20 |
- **CFG Parameter (eta)**: 0.1
|
|
|
|
| 14 |
## Model Details
|
| 15 |
|
| 16 |
- **Model Type**: flow_matching_mnist
|
| 17 |
+
- **Training Epochs**: 5000
|
| 18 |
- **Batch Size**: 250
|
| 19 |
- **Learning Rate**: 0.001
|
| 20 |
- **CFG Parameter (eta)**: 0.1
|