Open zzm422 opened 5 years ago
update line 176 turn 'x = transition(x, nb_filter, dropout_rate=dropout_rate, weight_decay=weight_decay)' into 'x = transition(x, concat_axis, nb_filter, dropout_rate=dropout_rate, weight_decay=weight_decay)'.
update line 176 turn 'x = transition(x, nb_filter, dropout_rate=dropout_rate, weight_decay=weight_decay)' into 'x = transition(x, concat_axis, nb_filter, dropout_rate=dropout_rate, weight_decay=weight_decay)'.