Skip to content
Extraits de code Groupes Projets
Valider c96c13b6 rédigé par Victor Joos de ter Beerst's avatar Victor Joos de ter Beerst
Parcourir les fichiers

fix visu

parent 1754774e
Aucune branche associée trouvée
Aucune étiquette associée trouvée
Aucune requête de fusion associée trouvée
......@@ -30,7 +30,9 @@ def main(cfg):
predict_datamodule = instantiate(cfg.predict_datamodule)
predictions = trainer.predict(model, datamodule=predict_datamodule, output="labels")
show_predictions(predict_datamodule.predict_dataset, predictions[0])
show_predictions(
predict_datamodule.predict_dataset, predictions[0], cfg.show, cfg.save
)
if __name__ == "__main__":
......
0% Chargement en cours ou .
You are about to add 0 people to the discussion. Proceed with caution.
Terminez d'abord l'édition de ce message.
Veuillez vous inscrire ou vous pour commenter