From c8cce325b1784b2743daaa8b7e467c3355513b5b Mon Sep 17 00:00:00 2001 From: Gabriel Humpire Date: Fri, 6 Mar 2020 10:21:54 +0100 Subject: [PATCH] Add tensorflow and pillow versions --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index eea13f0..11ac6cf 100644 --- a/README.md +++ b/README.md @@ -155,6 +155,8 @@ $ python train.py ../config/pascal_unet_res18_scse.yaml - pytorch: 1.0.0 - pretrainedmodels: 0.7.4 - albumentations: 0.1.8 +- tensorflow <= 1.15.2 +- pillow <= 6.2.2 if you want to train models in fp16 - NVIDIA/apex: 0.1