提交 c91cb6c5 编写于 作者: L Liufang Sang 提交者: whs

fix load checkpoint error in test_reader (#20924)

上级 925280b9
...@@ -13,7 +13,6 @@ ...@@ -13,7 +13,6 @@
# limitations under the license. # limitations under the license.
import os import os
import shutil
import paddle import paddle
import unittest import unittest
import paddle.fluid as fluid import paddle.fluid as fluid
...@@ -40,9 +39,6 @@ class TestReader(unittest.TestCase): ...@@ -40,9 +39,6 @@ class TestReader(unittest.TestCase):
return [('img', image.name), ('label', label.name)] return [('img', image.name), ('label', label.name)]
def quan(self, config_file): def quan(self, config_file):
if os.path.exists('./checkpoints_quan'):
shutil.rmtree('./checkpoints_quan')
if not fluid.core.is_compiled_with_cuda(): if not fluid.core.is_compiled_with_cuda():
return return
class_dim = 10 class_dim = 10
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册