fixed dogcat transfer

Author: lukas

keras-transfer/dogcat-transfer.py

@@ -110,7 +110,7 @@ def add_new_last_layer(base_model, nb_classes):
     epochs=nb_epoch,
     steps_per_epoch=nb_train_samples,
     validation_data=validation_generator,
-    callbacks=[WandbCallback(data_type="image", save_model=False)],
+    callbacks=[WandbCallback(data_type="image", labels=['cat', 'dog'],save_model=False)],
     class_weight='auto')