message: made a small mistake while decorating with Fixing that.
diff: " padding='same',\n",
  " kernel_initializer=initializer)\n",
  " \n",
  - " @tf.contrib.eager.defun()\n",
  + " @tf.contrib.eager.defun\n",
  " def call(self, x, training):\n",
  " # x shape == (bs, 256, 256, 3) \n",
  " x1 = self.down1(x, training=training) # (bs, 128, 128, 64)\n",
  " strides=1,\n",
  " kernel_initializer=initializer)\n",
  " \n",
  - " @tf.contrib.eager.defun()\n",
  + " @tf.contrib.eager.defun\n",
  " def call(self, inp, tar, training):\n",
  " # concatenating the input and the target\n",
  " x = tf.concat([inp, tar], axis=-1) # (bs, 256, 256, channels*2)\n",
  