Commit 76e78e9d authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent 128b3fcc
......@@ -82,7 +82,9 @@ tensorboard_callback = keras.callbacks.TensorBoard(log_dir=log_dir, histogram_fr
with open(
os.path.abspath(
train_path + "/" + [i for i in os.listdir(train_path) if i.endswith(".pickle")][0]
train_path
+ "/"
+ [i for i in os.listdir(train_path) if i.endswith(".pickle")][0]
),
"rb",
) as handle:
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment