From 72c82e6f02fc040bf875697ef0c591254fbc85b9 Mon Sep 17 00:00:00 2001 From: NilsDem Date: Wed, 4 Dec 2024 17:53:35 +0100 Subject: [PATCH] readme update --- README.md | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/README.md b/README.md index 25aa6bd..caedacd 100644 --- a/README.md +++ b/README.md @@ -9,6 +9,13 @@ We are currently working on a real-time implementation of this model called AFTE ## Model training + +Prior to training, install the required dependancies using : + +```bash +pip install -r "requirements.txt" +``` + Training the model requires three steps : processing the dataset, training an autoencoder, then training the diffusion model. ### Dataset preparation