Merge branch 'master' of github.com:iamhankai/attribute-aware-attention

iamhankai · Aug 31, 2019 · 4b4cf87 · 4b4cf87
2 parents 98acf35 + de958c0
commit 4b4cf87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CUB.py b/CUB.py
@@ -113,7 +113,7 @@ def load_data(data_folder, target_size=(224, 224), bounding_box=True):
 
 
 if __name__ == '__main__':
-    (X_train,y_train), (X_test,y_test) = load_data()
+    (X_train,y_train), (X_test,y_test),(A_train,A_test,C_A) = load_data()