Error on running train.py

[PR-Iyyer]

Using TensorFlow backend. train.py:41: UserWarning: Update your Conv2Dcall to the Keras 2 API:Conv2D(filters=32, input_shape=(1, 28, 28..., padding="same", kernel_size=(5, 5))model.add(Convolution2D(nb_filter=32,nb_row=5,nb_col=5,border_mode='same',input_shape=(1,28,28))) train.py:43: UserWarning: Update yourMaxPooling2Dcall to the Keras 2 API:MaxPooling2D(padding="same", pool_size=(2, 2))model.add(MaxPooling2D(pool_size=(2,2),border_mode='same')) train.py:44: UserWarning: Update yourConv2Dcall to the Keras 2 API:Conv2D(filters=64, input_shape=(32, 14, 1..., padding="same", kernel_size=(5, 5))model.add(Convolution2D(nb_filter=64,nb_row=5,nb_col=5,border_mode='same',input_shape=(32,14,14))) train.py:46: UserWarning: Update yourMaxPooling2Dcall to the Keras 2 API:MaxPooling2D(padding="same", pool_size=(2, 2))` model.add(MaxPooling2D(pool_size=(2,2),border_mode='same')) Traceback (most recent call last): File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/common_shapes.py", line 686, in _call_cpp_shape_fn_impl input_tensors_as_shapes, status) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/errors_impl.py", line 473, in exit c_api.TF_GetCode(self.status.status)) tensorflow.python.framework.errors_impl.InvalidArgumentError: Dimension 2 in both shapes must be equal, but are 28 and 32. Shapes are [5,5,28,32] and [5,5,32,1]. for 'Assign' (op: 'Assign') with input shapes: [5,5,28,32], [5,5,32,1].

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "train.py", line 55, in model.load_weights('model_weights.h5') File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/keras/models.py", line 737, in load_weights topology.load_weights_from_hdf5_group(f, layers) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/keras/engine/topology.py", line 3166, in load_weights_from_hdf5_group K.batch_set_value(weight_value_tuples) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/keras/backend/tensorflow_backend.py", line 2365, in batch_set_value assign_op = x.assign(assign_placeholder) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/ops/variables.py", line 594, in assign return state_ops.assign(self._variable, value, use_locking=use_locking) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/ops/state_ops.py", line 276, in assign validate_shape=validate_shape) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/ops/gen_state_ops.py", line 59, in assign use_locking=use_locking, name=name) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper op_def=op_def) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 3162, in create_op compute_device=compute_device) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 3208, in _create_op_helper set_shapes_for_outputs(op) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 2427, in set_shapes_for_outputs return _set_shapes_for_outputs(op) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 2400, in _set_shapes_for_outputs shapes = shape_func(op) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 2330, in call_with_requiring return call_cpp_shape_fn(op, require_shape_fn=True) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/common_shapes.py", line 627, in call_cpp_shape_fn require_shape_fn) File "/data/anaconda3/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/common_shapes.py", line 691, in _call_cpp_shape_fn_impl raise ValueError(err.message) ValueError: Dimension 2 in both shapes must be equal, but are 28 and 32. Shapes are [5,5,28,32] and [5,5,32,1]. for 'Assign' (op: 'Assign') with input shapes: [5,5,28,32], [5,5,32,1]. ` This is the error i got while running train.py. Can someone please help me to fix it?