Update train.ipynb

train.ipynb

@@ -295,7 +295,7 @@
    },
    "outputs": [],
    "source": [
-    "def train_network(iteration, loss_weight = [0.0, 0.0, 0.0001], check_iter = 1, test_iter = 10):\n",
+    "def train_network(iteration, loss_weight = [1.0, 100.0, 0.001], check_iter = 1, test_iter = 10):\n",
     "  for param in style_transfer_network.encoder.parameters():\n",
     "    # freeze parameter in the encoder network\n",
     "    param.requires_grad = False\n",