提交 aa7f8ed7 编写于 作者: X xixiaoyao

fix cls

上级 c877f4b5
export CUDA_VISIBLE_DEVICES=4
export CUDA_VISIBLE_DEVICES=3
python run.py
......@@ -71,7 +71,7 @@ class ClassifyReader(BaseReader):
self._batch_size = batch_size
self._num_epochs = num_epochs
self._data_generator = self._reader.data_generator( \
input_file, batch_size, num_epochs if phase == 'train' else 1, \
input_file, batch_size, num_epochs if self._phase == 'train' else 1, \
shuffle=shuffle_train if self._phase == 'train' else False, \
phase=self._phase)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册