Update README.md

bonlime · May 7, 2018 · 298d974 · 298d974
1 parent e08ffec
commit 298d974
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -33,4 +33,7 @@ How to use this model with custom input shape and custom number of classes:
 
 After that you will get a usual Keras model which you can train using .fit and .fit_generator methods
 
-
+I'm using 
+Keras==2.1.5
+tensorflow-gpu==1.6.0
+CUDA==9.0