Upload model README
Browse files
README.md
CHANGED
|
@@ -14,15 +14,14 @@ Flow Matching model with MLP vector field
|
|
| 14 |
## Model Details
|
| 15 |
|
| 16 |
- **Model Type**: flow_matching_mlp
|
| 17 |
-
- **Training Epochs**:
|
| 18 |
-
- **Batch Size**:
|
| 19 |
- **Learning Rate**: 0.001
|
| 20 |
|
| 21 |
## Architecture
|
| 22 |
|
| 23 |
-
- **Hidden Layers**: [
|
| 24 |
- **Data Dimensionality**: 2
|
| 25 |
-
- **Number of Modes**: 5
|
| 26 |
- **Standard Deviation**: 1.0
|
| 27 |
- **Scale**: 10.0
|
| 28 |
|
|
|
|
| 14 |
## Model Details
|
| 15 |
|
| 16 |
- **Model Type**: flow_matching_mlp
|
| 17 |
+
- **Training Epochs**: 20000
|
| 18 |
+
- **Batch Size**: 2000
|
| 19 |
- **Learning Rate**: 0.001
|
| 20 |
|
| 21 |
## Architecture
|
| 22 |
|
| 23 |
+
- **Hidden Layers**: [100, 100, 100, 100]
|
| 24 |
- **Data Dimensionality**: 2
|
|
|
|
| 25 |
- **Standard Deviation**: 1.0
|
| 26 |
- **Scale**: 10.0
|
| 27 |
|