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

Added missing argument of num_epochs

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-11 12:06:22 +02:00
parent 057bffcae3
commit a4e6ac2853
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

@ -62,7 +62,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
scenenet_data, nr_digits, length_dataset = \ scenenet_data, nr_digits, length_dataset = \
data.load_scenenet_data(file_names_photos, instances, args.coco_path, data.load_scenenet_data(file_names_photos, instances, args.coco_path,
batch_size=batch_size, batch_size=batch_size, num_epochs=args.num_epochs,
resized_shape=(image_size, image_size), resized_shape=(image_size, image_size),
mode="training") mode="training")
del file_names_photos, instances del file_names_photos, instances