Skip to content

Commit

Permalink
fix abn and dataset
Browse files Browse the repository at this point in the history
  • Loading branch information
zhizhangxian committed Oct 8, 2019
1 parent 5c4d4df commit aeb25d9
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
3 changes: 2 additions & 1 deletion config_utils/search_args.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,8 @@ def obtain_search_args():
parser.add_argument('--no-val', action='store_true', default=False,
help='skip validation during training')

parser.add_argument('--use-ABN', default=True, type=bool, help='whether use ABN')
parser.add_argument('--use-ABN', default=False, type=bool, help='whether use ABN')
parser.add_argument('--affine', default=False, type=bool, help='whether use affine in BN')
parser.add_argument('--dist', default=False, type=bool, help='whether use dist')
args = parser.parse_args()
return args
3 changes: 1 addition & 2 deletions mypath.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,7 @@ def db_root_dir(dataset):
elif dataset == 'sbd':
return '/path/to/datasets/benchmark_RELEASE/' # folder that contains dataset/.
elif dataset == 'cityscapes':
# return '/data/deeplearning/cityscapes/' # foler that contains leftImg8bit/
return r'E:\BaiduNetdiskDownload\cityscapes' # foler that contains leftImg8bit/
return '/data/deeplearning/cityscapes/' # foler that contains leftImg8bit/
elif dataset == 'kd':
return '/data/deeplearning/cityscapes/'
elif dataset == 'coco':
Expand Down

0 comments on commit aeb25d9

Please sign in to comment.