未验证 提交 2b706ca9 编写于 作者: G gushiqiao 提交者: GitHub

Fixed growing memory usage (#1469)

上级 d9fa688a
......@@ -383,8 +383,9 @@ class ReconstructionQuanter(object):
self._exe.run(startup_program)
start_time = time.time()
prev_start_time = start_time
loader = self._data_loader()
for epoch in range(self._epochs):
for i, data in enumerate(self._data_loader()):
for i, data in enumerate(loader):
prev_start_time = start_time
start_time = time.time()
out = self._exe.run(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册