weidai00 commited on
Commit
62977e4
·
verified ·
1 Parent(s): f0ab20b

Update AV/config/config_test_general.py

Browse files
Files changed (1) hide show
  1. AV/config/config_test_general.py +1 -1
AV/config/config_test_general.py CHANGED
@@ -104,7 +104,7 @@ def set_dataset(name):
104
  patch_size = patch_size_list[2]
105
 
106
  # path for dataset
107
- if dataset_name == 'DRIVE' or dataset_name == 'LES' or dataset_name == 'hrf':
108
  stride_height = 50
109
  stride_width = 50
110
  elif dataset_name == 'ukbb' or dataset_name == 'hrf':
 
104
  patch_size = patch_size_list[2]
105
 
106
  # path for dataset
107
+ if dataset_name == 'DRIVE' or dataset_name == 'LES':
108
  stride_height = 50
109
  stride_width = 50
110
  elif dataset_name == 'ukbb' or dataset_name == 'hrf':