From 89b96be2f5449f8f9cbca6ff030d83e6009a1b7b Mon Sep 17 00:00:00 2001 From: Mathias Gruber Date: Fri, 1 Mar 2019 13:39:29 +0100 Subject: [PATCH] Not using multiple workers right now --- main.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/main.py b/main.py index ba55bb1..36ecff3 100644 --- a/main.py +++ b/main.py @@ -25,7 +25,7 @@ # Sample call r""" # Train on CelebaHQ -python main.py --name CelebHQ --train C:\Documents\Kaggle\celebaHQ-512\train\ --validation C:\Documents\Kaggle\celebaHQ-512\val\ --test C:\Documents\Kaggle\celebaHQ-512\test\ " +python main.py --name CelebHQ --train C:\Documents\Kaggle\celebaHQ-512\train\ --validation C:\Documents\Kaggle\celebaHQ-512\val\ --test C:\Documents\Kaggle\celebaHQ-512\test\ --checkpoint "C:\Users\Mathias Felix Gruber\Documents\GitHub\PConv-Keras\data\logs\imagenet_phase1_paperMasks\weights.35-0.70.h5" """ @@ -62,12 +62,6 @@ def parse_args(): type=str, default='myDataset', help='Dataset name, e.g. \'imagenet\'' ) - - parser.add_argument( - '-workers', '--workers', - type=int, default=4, - help='How many workers to user for pre-processing' - ) parser.add_argument( '-batch_size', '--batch_size',