Skip to content

Commit b20c052

Browse files
committed
update README.md, fix issue with dropout
1 parent d04949f commit b20c052

File tree

3 files changed

+3
-5
lines changed

3 files changed

+3
-5
lines changed

README.md

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,7 @@ The goal of this tool is to provide a quick and easy way to execute Keras models
1616
- **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.
1717
- Batch prediction untested
1818
- BatchNormalization
19-
- Tested and working with:
20-
- Dense
21-
- GRU
19+
- Works with all supported layers
2220
- Activations:
2321
- ReLU
2422
- Sigmoid

konverter/utils/konverter_support.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,8 @@ def get_layer_info(self, layer):
109109

110110
try:
111111
wb = layer.get_weights()
112-
layer_class.info.has_weights = True # TODO: test dropout with this, TODO: unused, delete?
112+
if len(wb) == 0:
113+
return layer_class
113114
except:
114115
return layer_class
115116

konverter/utils/model_attributes.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,6 @@ class BaseModelInfo:
116116
class BaseLayerInfo:
117117
supported = False
118118
has_activation = False
119-
has_weights = False
120119
returns_sequences = False
121120
is_recurrent = False
122121
is_ignored = False

0 commit comments

Comments
 (0)