Hugging Face's logo

derekwong
/
flow-matching-mlp

PyTorch
flow_matching_mlp
diffusion
flow-matching
mlp
gaussian-mixture
Model card Files
xet
Community
derekwong commited on
Commit
68fb3ea
·
verified ·
1 Parent(s): 8702d1a

Upload model README

Browse files
Files changed (1) hide show
  1. README.md +3 -4
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**: 5000
18
- - **Batch Size**: 250
19
  - **Learning Rate**: 0.001
20
 
21
  ## Architecture
22
 
23
- - **Hidden Layers**: [64, 64, 64, 64]
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