How to

Lancer le container

docker pull registry.forgemia.inra.fr/remi.cresson/deepmapper_augmentation/augmentation:3.4-cpu
docker run -ti --rm registry.forgemia.inra.fr/remi.cresson/deepmapper_augmentation/augmentation:3.4-cpu bash

Telecharger le dataset "splitté"

wget https://nextcloud.inrae.fr/s/X5EWF48xgR6F5XS/download
unzip download
rm download

Lance l'apprentissage

python /app/train.py --tfrecords_dir /home/otbuser/patches_v5_splits/ --savedmodel /tmp/savedmodel

Note:

• il faut monter un dossier /logs sinon ça plante
• il faut donner le parametre savedmodel
• l'objectif c'est que MLFlow trace les métriques sur les données test (pas sur les données de validation). Le truc c'est que je ne suis plus certain que ces métriques soient explicitement logguées (pour le dataset "test").