master
ml server 2020-01-26 19:59:06 -07:00
parent 3442bffdf2
commit 0865b28c66
1 changed files with 4 additions and 4 deletions

View File

@ -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"