frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-28 10:34:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Pass loop_over optional parameter for cached reader properly. (#21929)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21929

Just need to pass `loop_over` argument properly.

Reviewed By: noname01

Differential Revision: D15885401

fbshipit-source-id: f1928277262a80e5b41f4c4f3945c2f378a4e233
This commit is contained in:
Zhanibek Datbayev
2019-06-19 18:12:15 -07:00
committed by Facebook Github Bot
parent 96c0bd3722
commit 4fee532de6
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
caffe2/python/dataio_test.py
View File

@ -364,7 +364,7 @@ class TestDBFileReader(TestCase):
# Read data for the first time.
cached_reader1 = CachedReader(
self._build_source_reader(ws, 100), db_path,
self._build_source_reader(ws, 100), db_path, loop_over=False,
)
build_cache_step = cached_reader1.build_cache_step()
session.run(build_cache_step)