update README.md, fix issue with dropout

sshane · May 8, 2020 · b20c052 · b20c052
1 parent d04949f
commit b20c052
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -16,9 +16,7 @@ The goal of this tool is to provide a quick and easy way to execute Keras models
     - **Important:** The current GRU support is based on [`GRU v3`](https://www.tensorflow.org/api_docs/python/tf/keras/layers/GRU) in tf.keras 2.1.0. It will not work correctly with older versions of TensorFlow.
     - Batch prediction untested
   - BatchNormalization
-    - Tested and working with:
-      - Dense
-      - GRU
+    - Works with all supported layers
 - Activations:
   - ReLU
   - Sigmoid

diff --git a/konverter/utils/konverter_support.py b/konverter/utils/konverter_support.py
@@ -109,7 +109,8 @@ def get_layer_info(self, layer):
 
     try:
       wb = layer.get_weights()
-      layer_class.info.has_weights = True  # TODO: test dropout with this, TODO: unused, delete?
+      if len(wb) == 0:
+        return layer_class
     except:
       return layer_class
 

diff --git a/konverter/utils/model_attributes.py b/konverter/utils/model_attributes.py
@@ -116,7 +116,6 @@ class BaseModelInfo:
 class BaseLayerInfo:
   supported = False
   has_activation = False
-  has_weights = False
   returns_sequences = False
   is_recurrent = False
   is_ignored = False