Print also a summary of a model for yolo tiny

girishmk0602 · May 20, 2020 · 3f8ed00 · 3f8ed00
1 parent 713aec7
commit 3f8ed00
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/save_model.py b/save_model.py
@@ -43,7 +43,7 @@ def save_tf():
       print("model option can be only 'yolov3' or 'yolov4'.")
       return
 
-    model.summary()
+  model.summary()
 
   model.save(FLAGS.output)