Commit a57fa460 authored by meak's avatar meak Committed by Yuxin Wu

fix: train_or_test to name (#1161)

parent 33efcb82
......@@ -50,7 +50,7 @@ class Caltech101Silhouettes(RNGDataFlow):
self.data = get_images_and_labels("caltech101_silhouettes_28_split1.mat")
if self.train_or_test == "train":
if self.name == "train":
self.images = self.data["train_data"].reshape((4100, 28, 28))
self.labels = self.data["train_labels"].ravel() - 1
elif self.name == "test":
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment