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

Update correct batch size

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-20 14:08:20 +02:00
parent 78ab277ce7
commit 8678f14083
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -46,7 +46,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
tf.enable_eager_execution() tf.enable_eager_execution()
batch_size = 16 batch_size = 32
image_size = 300 image_size = 300
use_dropout = False if args.network == "ssd" else True use_dropout = False if args.network == "ssd" else True
@ -118,7 +118,7 @@ def _auto_encoder_train(args: argparse.Namespace) -> None:
tf.enable_eager_execution() tf.enable_eager_execution()
coco_path = args.coco_path coco_path = args.coco_path
category = args.category category = args.category
batch_size = 32 batch_size = 16
image_size = 256 image_size = 256
coco_data = data.load_coco_train(coco_path, category, num_epochs=args.num_epochs, batch_size=batch_size, coco_data = data.load_coco_train(coco_path, category, num_epochs=args.num_epochs, batch_size=batch_size,
resized_shape=(image_size, image_size)) resized_shape=(image_size, image_size))