diff --git a/modules/deeplearning/cloudheight_mgpus.py b/modules/deeplearning/cloudheight_mgpus.py index 49330f5b38a59f70a41dab057634614436c4b655..fe4d3d5c18068af7803d6b84c9053fda41282f83 100644 --- a/modules/deeplearning/cloudheight_mgpus.py +++ b/modules/deeplearning/cloudheight_mgpus.py @@ -18,7 +18,7 @@ DISK_CACHE = True PROC_BATCH_SIZE = 60 NumLabels = 1 -BATCH_SIZE = 1024 +BATCH_SIZE = 256 NUM_EPOCHS = 200 GLOBAL_BATCH_SIZE = BATCH_SIZE * 3 PROC_BATCH_BUFFER_SIZE = 50000 @@ -40,11 +40,11 @@ abi_channels = abi_2km_channels abi_mean = {'08': 236.014, '14': 275.229, '02': 0.049, '11': 273.582, '13': 275.796, '15': 272.928, '16': 260.956, '09': 244.502, '10': 252.375} abi_std = {'08': 7.598, '14': 20.443, '02': 0.082, '11': 19.539, '13': 20.431, '15': 20.104, '16': 15.720, '09': 9.827, '10': 11.765} abi_valid_range = {'02': [0.001, 120], '08': [150, 350], '14': [150, 350], '11': [150, 350], '13': [150, 350], '15': [150, 350], '16': [150, 350], '09': [150, 350], '10': [150, 350]} -#abi_half_width = {'08': 12, '14': 12, '02': 48, '11': 12, '13': 12, '15': 12, '16': 12, '09': 12, '10': 12} -abi_half_width = {'08': 3, '14': 3, '02': 12, '11': 3, '13': 3, '15': 3, '16': 3, '09': 3, '10': 3} +abi_half_width = {'08': 12, '14': 12, '02': 48, '11': 12, '13': 12, '15': 12, '16': 12, '09': 12, '10': 12} +#abi_half_width = {'08': 3, '14': 3, '02': 12, '11': 3, '13': 3, '15': 3, '16': 3, '09': 3, '10': 3} abi_stride = {'08': 1, '14': 1, '02': 4, '11': 1, '13': 1, '15': 1, '16': 1, '09': 1, '10': 1} -#img_width = 24 -img_width = 6 +img_width = 24 +#img_width = 6 NUM_VERT_LEVELS = 26 NUM_VERT_PARAMS = 2