Skip to content

Commit 3041fa3

Browse files
authored
Merge pull request #292 from bensussman/shuffle-data
Shuffle self.data after assignment
2 parents 60aa97b + 890694a commit 3041fa3

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

model.py

+2
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@ def __init__(self, sess, input_height=108, input_width=108, crop=True,
7676
self.c_dim = self.data_X[0].shape[-1]
7777
else:
7878
self.data = glob(os.path.join(self.data_dir, self.dataset_name, self.input_fname_pattern))
79+
np.random.shuffle(self.data)
7980
imreadImg = imread(self.data[0])
8081
if len(imreadImg.shape) >= 3: #check if image is a non-grayscale image by checking channel number
8182
self.c_dim = imread(self.data[0]).shape[-1]
@@ -194,6 +195,7 @@ def train(self, config):
194195
else:
195196
self.data = glob(os.path.join(
196197
config.data_dir, config.dataset, self.input_fname_pattern))
198+
np.random.shuffle(self.data)
197199
batch_idxs = min(len(self.data), config.train_size) // config.batch_size
198200

199201
for idx in xrange(0, batch_idxs):

0 commit comments

Comments
 (0)