Commit 0e0acada authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent 76e78e9d
......@@ -81,10 +81,8 @@ log_dir = os.path.abspath(
tensorboard_callback = keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1)
with open(
os.path.abspath(
train_path
+ "/"
+ [i for i in os.listdir(train_path) if i.endswith(".pickle")][0]
os.path.join(
train_path, [i for i in os.listdir(train_path) if i.endswith(".pickle")][0]
),
"rb",
) as handle:
......
......@@ -150,10 +150,8 @@ else:
tensorboard_callback = keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1)
with open(
os.path.abspath(
train_path
+ "/"
+ [i for i in os.listdir(train_path) if i.endswith(".pickle")][0]
os.path.join(
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