Skip to content

Commit

Permalink
cosmetic fix
Browse files Browse the repository at this point in the history
  • Loading branch information
TimSalimans authored and TimSalimans committed Dec 3, 2016
1 parent e62dd22 commit 2346cfd
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions keras/weightnorm.py
Original file line number Diff line number Diff line change
Expand Up @@ -181,10 +181,10 @@ def add_weightnorm_param_updates(updates, new_V_param, new_g_param, W, V_scaler)
# data based initialization for a given Keras model
def data_based_init(model, input):

# input can be feed_dict, numpy array, or list of numpy arrays
# input can be dict, numpy array, or list of numpy arrays
if type(input) is dict:
feed_dict = input
if type(input) is list:
elif type(input) is list:
feed_dict = {tf_inp: np_inp for tf_inp,np_inp in zip(model.inputs,input)}
else:
feed_dict = {model.inputs[0]: input}
Expand Down

0 comments on commit 2346cfd

Please sign in to comment.