Skip to content

S
seminar-breakout
Commit a57fa460 authored by meak's avatar meak Committed by Yuxin Wu
Browse files
Options

fix: train_or_test to name (#1161)

parent 33efcb82
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/dataflow/dataset/caltech101.py
View file @ a57fa460
...@@ -50,7 +50,7 @@ class Caltech101Silhouettes(RNGDataFlow): ...@@ -50,7 +50,7 @@ class Caltech101Silhouettes(RNGDataFlow):
self.data = get_images_and_labels("caltech101_silhouettes_28_split1.mat") 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.images = self.data["train_data"].reshape((4100, 28, 28))
self.labels = self.data["train_labels"].ravel() - 1 self.labels = self.data["train_labels"].ravel() - 1
elif self.name == "test": 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