added saved_model to allow loading from with huggingface hub pretrained models
Browse files- saved_model.pb +3 -0
- utils/h5_to_saved_pb.py +6 -0
- variables/variables.data-00000-of-00001 +2 -2
- variables/variables.index +0 -0
saved_model.pb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:322fc851fd1221bf42dea330f2f93be3025e958b505ab51eda8c2ae86f7ab47a
|
| 3 |
+
size 3499342
|
utils/h5_to_saved_pb.py
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import tensorflow as tf
|
| 2 |
+
|
| 3 |
+
|
| 4 |
+
model = tf.keras.models.load_model("tf_model.h5")
|
| 5 |
+
|
| 6 |
+
tf.saved_model.save(model, "")
|
variables/variables.data-00000-of-00001
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d6d4a304371ff904a5863dcdbbb6527fa10ff9e454d9692dff80312008ff6e60
|
| 3 |
+
size 63451732
|
variables/variables.index
CHANGED
|
Binary files a/variables/variables.index and b/variables/variables.index differ
|
|
|