Added callback

vanpelt · vanpelt · commit fa47c75a84c5 · 2019-09-17T17:44:15.000-07:00
diff --git a/examples/keras-sweep/notebook.ipynb b/examples/keras-sweep/notebook.ipynb
@@ -35,7 +35,7 @@
     "                  loss='sparse_categorical_crossentropy',\n",
     "                  metrics=['accuracy'])\n",
     "    \n",
-    "    model.fit(train_images, train_labels, epochs=5,\n",
+    "    model.fit(train_images, train_labels, epochs=5, callbacks=[wandb.keras.WandbCallback(input_type=\"images\", save_model=False)],\n",
     "                  validation_data=(test_images, test_labels))"
    ]
   },

Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,7 @@`
`35`	`35`	`" loss='sparse_categorical_crossentropy',\n",`
`36`	`36`	`" metrics=['accuracy'])\n",`
`37`	`37`	`" \n",`
`38`		`- " model.fit(train_images, train_labels, epochs=5,\n",`
	`38`	`+ " model.fit(train_images, train_labels, epochs=5, callbacks=[wandb.keras.WandbCallback(input_type=\"images\", save_model=False)],\n",`
`39`	`39`	`" validation_data=(test_images, test_labels))"`
`40`	`40`	`]`
`41`	`41`	`},`