diff --git a/utils/loaders.py b/utils/loaders.py index 27c4253397..9a36df3139 100644 --- a/utils/loaders.py +++ b/utils/loaders.py @@ -35,7 +35,7 @@ def build(self, att, batch_size, label = None): , x_col='image_id' , y_col=label , target_size=self.target_size - , class_mode='other' + , class_mode='raw' , batch_size=batch_size , shuffle=True )