Update README.md

README.md

@@ -10,14 +10,14 @@ language:
 
 ### Model Description
 
-<!-- Provide a longer summary of what this model is. -->
+This repo includes every models we trained during the Jax Community event sprint, organized by Hugging Face.
+The folders {model} contains the Flax checkpoint and {model}_pt the Torch checkpoint.
 
 
 
 - **Developed by:** Baptiste Lemaire, Guillaume Thomas and Tom Dupuis from CEA-List
 - **Model type:** Canny Edge Maps conditionned Diffusion model 
 - **Language(s) (NLP):** English
-- **License:** [More Information Needed]
 
 
 ## Uses
@@ -36,6 +36,8 @@ See our gradio app for more information : [UCDR-Net gradio](https://huggingface.
 * [Bridge](https://sites.google.com/view/bridgedata)
 
 ### Training Procedure 
+We trained from scratch each one of our models. We kept the initial parameters, except for the Batch Size.
+You can find the training script in the following [Event repo's folder](https://github.com/huggingface/community-events/blob/main/jax-controlnet-sprint/training_scripts/train_controlnet_flax.py)
 
 
 #### Preprocessing 
@@ -44,9 +46,16 @@ See our gradio app for more information : [UCDR-Net gradio](https://huggingface.
 -Canny Edge Map
 
 
-#### Training Hyperparameters and Speeds, Sizes, Times
+#### Training parameters
+The following table describes the differents hyperpa
+![alt text](./table_training.png)
+
+We stopped the coyo model a bit after it processed its first epoch. After running it, we discovered it performed pretty well even after only one epoch. So we deciced to keep it.
+
+
+The last model has been trained with a custom DataLoader. The previous loads a batch containing 4 images from Bridge and 28 from Coyo.
+Therefore, we can't talk about epoch as the model processed coyo faster than bridge. We then trained the model according to steps and not epoch. 
 
-See README.md file in model folders ({model}, not {model}_pt)
 
 
 ### Results
@@ -68,11 +77,4 @@ Carbon emissions can be estimated using the [Machine Learning Impact calculator]
 - **Carbon Emitted:** [More Information Needed]
 
 
-## Citation [optional]
-
-<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
-
-**BibTeX:**
-
-[More Information Needed]