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

Resized images to 64x64 to alleviate memory problems

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-11 15:20:28 +02:00
parent 41fe940d5b
commit e7be92550d
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/twomartens/masterthesis/main.py
View File

@ -103,7 +103,8 @@ def _auto_encoder_train(args: argparse.Namespace) -> None:
coco_path = args.coco_path coco_path = args.coco_path
category = args.category category = args.category
batch_size = 32 batch_size = 32
coco_data = data.load_coco(coco_path, category, num_epochs=args.num_epochs, batch_size=batch_size) coco_data = data.load_coco(coco_path, category, num_epochs=args.num_epochs, batch_size=batch_size,
resized_shape=(64, 64))
train_summary_writer = summary_ops_v2.create_file_writer( train_summary_writer = summary_ops_v2.create_file_writer(
f"{args.summary_path}/train/category-{category}/{args.iteration}" f"{args.summary_path}/train/category-{category}/{args.iteration}"
) )