Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
未验证 提交 9ee1dc53 编写于 作者: N niuliling123 提交者： GitHub
浏览文件
操作

update (#42248)

上级 1bf08eca
显示空白变更内容
内联 并排
Showing with 2 addition and 1 deletion
python/paddle/fluid/reader.py
浏览文件 @ 9ee1dc53
...@@ -19,6 +19,7 @@ import numpy as np ...@@ -19,6 +19,7 @@ import numpy as np
import threading import threading
import paddle import paddle
import time import time
import copy
from .framework import Program, Variable, program_guard, default_main_program, default_startup_program, _non_static_mode, cpu_places, _current_expected_place, _in_eager_without_dygraph_check from .framework import Program, Variable, program_guard, default_main_program, default_startup_program, _non_static_mode, cpu_places, _current_expected_place, _in_eager_without_dygraph_check
from .executor import global_scope from .executor import global_scope
...@@ -214,7 +215,7 @@ class AuToTune(object): ...@@ -214,7 +215,7 @@ class AuToTune(object):
return sub_dataset return sub_dataset
def get_autotune_loader(self): def get_autotune_loader(self):
loader = self.loader loader = copy.copy(self.loader)
batch_size = self.loader.batch_sampler.batch_size batch_size = self.loader.batch_sampler.batch_size
if isinstance(self.loader.batch_sampler, if isinstance(self.loader.batch_sampler,
paddle.io.DistributedBatchSampler): paddle.io.DistributedBatchSampler):
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或