Truncate keras summary at 10k characters

DLHub-Argonne · Jul 25, 2019 · aaab524 · aaab524
1 parent 87e55df
commit aaab524
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/dlhub_sdk/models/servables/keras.py b/dlhub_sdk/models/servables/keras.py
@@ -7,7 +7,7 @@
 from dlhub_sdk.utils.types import compose_argument_block
 
 _keras_version_tuple = tuple(int(i) for i in keras_version.split("."))
-
+_summary_limit = 10000
 
 def _detect_backend(output):
     """Add the backend
@@ -82,13 +82,16 @@ def create_model(cls, model_path, output_names=None, arch_path=None,
         if output_names is not None:
             output['servable']['methods']['run']['method_details']['classes'] = output_names
 
-        # Get a full description of the model
+        # Get a full description of the model. Limit summary to _summary_limit in length
         output.summary = ""
 
         def capture_summary(x):
             output.summary += x + "\n"
 
         model.summary(print_fn=capture_summary)
+        output.summary = (output.summary[:_summary_limit] + '<<TRUNCATED>>') \
+                            if  len(output.summary) > _summary_limit else output.summary
+
         output['servable']['model_summary'] = output.summary
         output['servable']['model_type'] = 'Deep NN'