derekwong
/

flow-matching-mlp

flow_matching_mlp

gaussian-mixture

Model card Files Files and versions

derekwong commited on 13 days ago

Commit

8fcee03

·

verified ·

1 Parent(s): b25e854

Upload model README

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ Flow Matching model with MLP vector field
 ## Model Details
 - **Model Type**: flow_matching_mlp
-- **Training Epochs**: 100
 - **Batch Size**: 250
 - **Learning Rate**: 0.001

 ## Model Details
 - **Model Type**: flow_matching_mlp
+- **Training Epochs**: 5000
 - **Batch Size**: 250
 - **Learning Rate**: 0.001