From bed06eb99cb1eabd815ce567d2e7ba0423dc0432 Mon Sep 17 00:00:00 2001
From: Sting <loic.allegre@ensiie.fr>
Date: Fri, 23 Jun 2023 11:50:55 +0200
Subject: [PATCH] Update README

---
 README.md | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index f936290..d22d58f 100644
--- a/README.md
+++ b/README.md
@@ -77,9 +77,19 @@ $ ./process_train_data video_folder train_folder
 
 To prepare the training data for the model :
 ```bash
-$ python music_processor.py train train_folder 
+$ python cnn_prepare_data.py train train_folder 
 ```
 
+Prepared data will be stored in `./data/pickles/train_data.pickle`
+
+To train the model on the prepared data :
+```bash
+$ python cnn_train.py
+```
+
+The model will be written to `./models/model.pth`
+
+
 ## Infer
 
 To execute AutoKara on a MKV video file :
-- 
GitLab