updated notebook

notebooks/enhance_me_train.ipynb

@@ -60,6 +60,7 @@
     "apply_random_horizontal_flip = True  # @param {type:\"boolean\"}\n",
     "apply_random_vertical_flip = True  # @param {type:\"boolean\"}\n",
     "apply_random_rotation = True  # @param {type:\"boolean\"}\n",
+    "use_mixed_precision = True  # @param {type:\"boolean\"}\n",
     "wandb_api_key = \"\"  # @param {type:\"string\"}\n",
     "val_split = 0.1  # @param {type:\"slider\", min:0.1, max:1.0, step:0.1}\n",
     "batch_size = 4  # @param {type:\"integer\"}\n",
@@ -121,6 +122,7 @@
    "outputs": [],
    "source": [
     "mirnet.build_model(\n",
+    "    use_mixed_precision=use_mixed_precision,\n",
     "    num_recursive_residual_groups=num_recursive_residual_groups,\n",
     "    num_multi_scale_residual_blocks=num_multi_scale_residual_blocks,\n",
     "    learning_rate=learning_rate,\n",