add no curvature option

2023-01-26 12:03:50 +00:00
parent 2ac1a812fc
commit 44e1d02e5a
4 changed files with 32 additions and 19 deletions
@@ -311,7 +311,7 @@ def load_model(model, checkpoint, disable_assert=False, assert_consumed_on=False

 def get_summary_str(model):
    lines = []
-    model.summary(print_fn=lines.append, line_length=150,expand_nested=True,show_trainable=True)
+    model.summary(print_fn=lines.append, line_length=150, expand_nested=True, show_trainable=True)
    # Add initial spaces to avoid markdown formatting in TensorBoard
    return lines