loss function is actually reducing \!\!\! :) yay :)
"metadata": {},
"outputs": [],
"source": [
- "for parameters, img in test_dataset.take(1):\n",
+ "for parameters, img in create_pipeline_dataset(1).take(1):\n",
" results_figures(model, img, parameters)"
]
},
"metadata": {},
"outputs": [],
"source": [
+ "optimizer = tf.keras.optimizers.Adam()\n",
+ "\n",
"checkpoint_dir = './training_checkpoints'\n",
"checkpoint_prefix = os.path.join(checkpoint_dir, \"ckpt\")\n",
"checkpoint = tf.train.Checkpoint(optimizer=optimizer,\n",
"\n",
"test_dataset = create_pipeline_dataset(1)\n",
"train_dataset = create_pipeline_dataset(BATCH_SIZE)\n",
- "optimizer = tf.keras.optimizers.Adam(learning_rate=0.1, beta_1=0.5)\n",
+ "\n",
"\n",
"\n",
"@tf.function(experimental_relax_shapes=True)\n",
