diff --git a/CNN_code_one_gridcell b/CNN_code_one_gridcell index 5d1cdc1..c5424bb 100644 --- a/CNN_code_one_gridcell +++ b/CNN_code_one_gridcell @@ -77,7 +77,7 @@ cg_days = nldn_vec[:,cell_loc] learning_rate = np.load('learning_rates.npy')[cellnum-1] batch_size = int(np.load('batch_sizes.npy')[cellnum-1]) conv_filters = int(np.load('conv_filters.npy')[cellnum-1]) -dense_layers = int(np.load('dense_layers.npy')[cellnum-1]) +dense_layers = 1 dense_neurons = 16 activity_reg = np.load('activity_regs.npy')[cellnum-1] dropout_rate = np.load('dropout_rates.npy')[cellnum-1]