From d96585e514bef44e93b89e737e4906179cb4b106 Mon Sep 17 00:00:00 2001 From: ml server Date: Wed, 29 Jan 2020 20:15:41 -0700 Subject: [PATCH] train GMSK --- notebooks/wut-train.ipynb | 44 +++++++++------------------------------ 1 file changed, 10 insertions(+), 34 deletions(-) diff --git a/notebooks/wut-train.ipynb b/notebooks/wut-train.ipynb index c42a9be..d97aeda 100644 --- a/notebooks/wut-train.ipynb +++ b/notebooks/wut-train.ipynb @@ -149,7 +149,7 @@ "metadata": {}, "outputs": [], "source": [ - "ENCODING='PSK31'" + "ENCODING='GMSK'" ] }, { @@ -221,27 +221,11 @@ "print('total training good images:', num_train_good)\n", "print('total training bad images:', num_train_bad)\n", "print(\"--\")\n", - "print(\"Total training images:\", total_train)" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": {}, - "outputs": [], - "source": [ + "print(\"Total training images:\", total_train)\n", "print('total validation good images:', num_val_good)\n", "print('total validation bad images:', num_val_bad)\n", "print(\"--\")\n", - "print(\"Total validation images:\", total_val)" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": {}, - "outputs": [], - "source": [ + "print(\"Total validation images:\", total_val)\n", "print(\"Reduce training and validation set when testing\")\n", "#total_train = 100\n", "#total_val = 100\n", @@ -257,7 +241,7 @@ "metadata": {}, "outputs": [], "source": [ - "batch_size = 64\n", + "batch_size = 32\n", "epochs = 16" ] }, @@ -267,12 +251,8 @@ "metadata": {}, "outputs": [], "source": [ - "# Half size\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" + "IMG_WIDTH = 832\n", + "IMG_HEIGHT = 1606" ] }, { @@ -281,9 +261,7 @@ "metadata": {}, "outputs": [], "source": [ - "train_image_generator = ImageDataGenerator(\n", - " rescale=1./255\n", - ")" + "train_image_generator = ImageDataGenerator( rescale=1./255 )" ] }, { @@ -292,9 +270,7 @@ "metadata": {}, "outputs": [], "source": [ - "val_image_generator = ImageDataGenerator(\n", - " rescale=1./255\n", - ")" + "val_image_generator = ImageDataGenerator( rescale=1./255 )" ] }, { @@ -572,7 +548,7 @@ "metadata": {}, "outputs": [], "source": [ - "model.save('/srv/satnogs/data/models/PSK31/wut-PSK31-201912.h5')" + "model.save('/srv/satnogs/data/models/GMSK/wut-GMSK-201912.h5')" ] }, { @@ -581,7 +557,7 @@ "metadata": {}, "outputs": [], "source": [ - "model.save('/srv/satnogs/data/models/PSK31/wut-PSK31-201912.tf')" + "model.save('/srv/satnogs/data/models/GMSK/wut-GMSK-201912.tf')" ] }, {