Skip to content

Commit 435a96f

Browse files
committed
set clear devices True while saving model
1 parent 0dfb14a commit 435a96f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

freeze_yolo_tf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
ckpt_file = "./checkpoint/rdt_model" # Model to freeze
1818
output_node_names = ["input/input_data", "pred_sbbox/concat_2", "pred_mbbox/concat_2", "pred_lbbox/concat_2"]
1919
model_export_path = "./models/Flu_audere" # export directory
20-
version = 1
20+
version = 2
2121
export_path = os.path.join(model_export_path, str(version))
2222

2323

@@ -86,7 +86,7 @@
8686

8787
main_op=tf.tables_initializer(),
8888

89-
strip_default_attrs=True)
89+
strip_default_attrs=True,clear_devices=True)
9090

9191

9292

0 commit comments

Comments
 (0)