Skip to content

Commit 278b91f

Browse files
committed
Fixing --release option
1 parent 0448b14 commit 278b91f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

config.py

+2
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,8 @@ def test_steps(self) -> int:
171171
return ceil(self.NUM_TEST_EXAMPLES / self.TEST_BATCH_SIZE) if self.TEST_BATCH_SIZE else 0
172172

173173
def data_path(self, is_evaluating: bool = False):
174+
if self.RELEASE:
175+
return ''
174176
return self.TEST_DATA_PATH if is_evaluating else self.train_data_path
175177

176178
def batch_size(self, is_evaluating: bool = False):

0 commit comments

Comments
 (0)