Skip to content

Commit

Permalink
Not using multiple workers right now
Browse files Browse the repository at this point in the history
  • Loading branch information
MathiasGruber committed Mar 1, 2019
1 parent a6d5cfa commit 89b96be
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions main.py
Expand Up @@ -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"
"""


Expand Down Expand Up @@ -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',
Expand Down

0 comments on commit 89b96be

Please sign in to comment.