spacecruft
/
satnogs-wut
0
0
Fork 0
Code Issues 6 Releases Activity

Training cleanup, new mode

master
Jeff Moe 2022-08-16 21:03:02 -06:00
parent 323ceda7eb
commit 3068a39e3a
1 changed files with 38 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
notebooks/wut-train.ipynb
View File

@ -89,7 +89,7 @@
"metadata": {},
"outputs": [],
"source": [
"ENCODING='FM'\n",
"ENCODING='FSK9k6'\n",
"batch_size = 64\n",
"epochs = 4\n",
"# Failing with this now:\n",
@ -125,21 +125,18 @@
"metadata": {},
"outputs": [],
"source": [
"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)\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)\n",
"#print(\"Reduce training and validation set when testing\")\n",
"#total_train = 100\n",
"#total_val = 100\n",
"print(\"Train =\")\n",
"print(total_train)\n",
"print(\"Validation =\")\n",
"print(total_val)"
"print('Training good images: ', num_train_good)\n",
"print('Training bad images: ', num_train_bad)\n",
"print('Training images: ', total_train)\n",
"print('Validation good images: ', num_val_good)\n",
"print('Validation bad images: ', num_val_bad)\n",
"print('Validation images: ', total_val)\n",
"print('')\n",
"print('Reduce training and validation set')\n",
"total_train = 1000\n",
"total_val = 1000\n",
"print('Training reduced to: ', total_train)\n",
"print('Validation reduced to: ', total_val)"
]
},
{
@ -323,7 +320,7 @@
"metadata": {},
"outputs": [],
"source": [
"model.summary()"
"#model.summary()"
]
},
{
@ -444,7 +441,9 @@
"metadata": {},
"outputs": [],
"source": [
"model.save('/srv/satnogs/data/models/FM/wut-FM.h5')"
"h5_file=(\"wut-\" + ENCODING + \".h5\")\n",
"save_path_h5 = os.path.join('/srv/satnogs/data/models/', ENCODING, h5_file)\n",
"print(save_path_h5)"
]
},
{
@ -453,7 +452,27 @@
"metadata": {},
"outputs": [],
"source": [
"model.save('/srv/satnogs/data/models/FM/wut-FM.tf')"
"model.save(save_path_h5)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"tf_modeldir=(\"wut-\" + ENCODING + \".tf\")\n",
"save_path_tf = os.path.join('/srv/satnogs/data/models/', ENCODING, tf_modeldir)\n",
"print(save_path_tf)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"model.save(save_path_tf)"
]
},
{