Skip to content

Commit 80c2b01

Browse files
committed
Ignore .json files from profiler and fix typo "moodel"
1 parent c7b5a8e commit 80c2b01

File tree

2 files changed

+18
-3
lines changed

2 files changed

+18
-3
lines changed

.gitignore

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,4 +116,7 @@ ENV/
116116
# *.e*
117117

118118
# Etc
119-
*.local
119+
*.local
120+
121+
# TF Profiler output
122+
*.json

plasma/models/builder.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -373,17 +373,29 @@ def save_model_weights(self, model, epoch):
373373
signatures=None, # applicable to 'tf' SavedModel format only
374374
)
375375
# TensorFlow SavedModel format (full directory)
376-
full_moodel_save_dir = full_model_save_path.rsplit('.',1)[0]
376+
full_model_save_dir = full_model_save_path.rsplit('.',1)[0]
377377
# TODO(KGF): model.save(..., save_format='tf') disabled in r1.15
378378
# Same with tf.keras.models.save_model(..., save_format="tf").
379379
# Need to use experimental API until r2.x
380380
# model.save(full_model_save_dir, overwrite=True, save_format='tf')
381-
tf.keras.experimental.export_saved_model(model, full_moodel_save_dir,
381+
tf.keras.experimental.export_saved_model(model, full_model_save_dir,
382382
custom_objects=None,
383383
as_text=False,
384384
input_signature=None,
385385
serving_only=False
386386
)
387+
# WARNING:tensorflow:Export includes no default signature!
388+
389+
# KGF: is the above comment accurate? 1.15.0 on macOS marks export_saved_model()
390+
# as deprecated, but below save() call errors out
391+
# model.save(full_model_save_dir,
392+
# overwrite=True,
393+
# include_optimizer=True,
394+
# save_format='tf',
395+
# signatures=None,
396+
# )
397+
398+
387399
# try:
388400
if _has_onnx:
389401
save_path = self.get_save_path(epoch, ext='onnx')

0 commit comments

Comments
 (0)