Commit fdbfbb65 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent eab54849
......@@ -149,6 +149,7 @@ assert input_type in [
if hparams is not None:
with open(hparams, "rb") as handle:
hparams = pickle.load(handle)
hparams["encoding"] = encoding
else:
hparams = {
"units_conv": 256,
......
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