updated notebook

notebooks/enhance_me_train.ipynb

@@ -22,7 +22,7 @@
    },
    "outputs": [],
    "source": [
-    "!git clone https://github.com/soumik12345/enhance-me -b zero-dce\n",
+    "!git clone https://github.com/soumik12345/enhance-me\n",
     "!pip install -qqq wandb streamlit"
    ]
   },
@@ -187,16 +187,15 @@
    "source": [
     "# @title Zero-DCE Train Configs\n",
     "\n",
-    "experiment_name = \"lol_dataset_256_resize\"  # @param {type:\"string\"}\n",
+    "experiment_name = \"unpaired_low_light_256_resize\"  # @param {type:\"string\"}\n",
     "image_size = 256  # @param {type:\"integer\"}\n",
-    "dataset_label = \"lol\"  # @param [\"lol\"]\n",
+    "dataset_label = \"unpaired\"  # @param [\"lol\", \"unpaired\"]\n",
     "use_mixed_precision = False  # @param {type:\"boolean\"}\n",
     "apply_resize = True  # @param {type:\"boolean\"}\n",
     "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 = False  # @param {type:\"boolean\"}\n",
-    "wandb_api_key = \"8d7149fe07496df2aaab8e9856a6ed8564e2a644\"  # @param {type:\"string\"}\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 = 16  # @param {type:\"integer\"}\n",
     "learning_rate = 1e-4  # @param {type:\"number\"}\n",