Skip to content

Commit fb3a6d3

Browse files
authored
Add files via upload
Added hook.close()
1 parent 974028d commit fb3a6d3

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

notebooks/scripts/train_tf2_fashion_mnist_custom_sdp_debugger.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -381,4 +381,6 @@ def save_model(model, model_dir):
381381
evaluate_model(model, x_test, y_test)
382382
# Save the generated model
383383
save_model(model, args.model_dir)
384+
# Close the SMDebugger hook
385+
hook.close()
384386
logger.info('Completed executing the main() function.')

0 commit comments

Comments
 (0)