2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Pickle history property instead of History object

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-04 13:02:06 +02:00
parent 2b93480df9
commit d3be3ea5f9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/cli.py
View File

@ -154,7 +154,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
) )
with open(f"{args.summary_path}/train/{args.network}/{args.iteration}/history", "wb") as file: with open(f"{args.summary_path}/train/{args.network}/{args.iteration}/history", "wb") as file:
pickle.dump(history, file) pickle.dump(history.history, file)
def _auto_encoder_train(args: argparse.Namespace) -> None: def _auto_encoder_train(args: argparse.Namespace) -> None: