From 0865b28c66dea06508d9af700ab930663b73ced1 Mon Sep 17 00:00:00 2001 From: ml server Date: Sun, 26 Jan 2020 19:59:06 -0700 Subject: [PATCH] CW train --- notebooks/wut-train.ipynb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/notebooks/wut-train.ipynb b/notebooks/wut-train.ipynb index 99351d6..ec57bb2 100644 --- a/notebooks/wut-train.ipynb +++ b/notebooks/wut-train.ipynb @@ -275,8 +275,8 @@ "outputs": [], "source": [ "# Good results\n", - "batch_size = 256\n", - "epochs = 6\n", + "batch_size = 512\n", + "epochs = 8\n", "#\n", "# Large Test\n", "# With 220G RAM\n", @@ -310,8 +310,8 @@ "outputs": [], "source": [ "# Half size\n", - "IMG_HEIGHT = 416\n", - "IMG_WIDTH= 804\n", + "IMG_HEIGHT = 804\n", + "IMG_WIDTH= 416\n", "# Full size, machine barfs probably needs more RAM\n", "#IMG_HEIGHT = 832\n", "#IMG_WIDTH = 1606"