finetune labels

lukas · lukas · commit 20d53f10dc98 · 2018-07-27T08:44:51.000-07:00
diff --git a/keras-transfer/dogcat-finetune.py b/keras-transfer/dogcat-finetune.py
@@ -118,7 +118,7 @@ def setup_to_finetune(model):
     steps_per_epoch=2,
     epochs=20,
     validation_data = val_data,
-    callbacks=[WandbCallback(data_type="image")],
+    callbacks=[WandbCallback(data_type="image", labels=["cat", "dog"])],
     )
 
 model.save(args.output_model_file)

Original file line number	Diff line number	Diff line change
`@@ -118,7 +118,7 @@ def setup_to_finetune(model):`
`118`	`118`	`steps_per_epoch=2,`
`119`	`119`	`epochs=20,`
`120`	`120`	`validation_data = val_data,`
`121`		`- callbacks=[WandbCallback(data_type="image")],`
	`121`	`+ callbacks=[WandbCallback(data_type="image", labels=["cat", "dog"])],`
`122`	`122`	`)`
`123`	`123`
`124`	`124`	`model.save(args.output_model_file)`