Skip to content

S
seminar-breakout
Commit bf683cd2 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

[FasterRCNN] Disable prefetch for now (fix #546)

parent a8e9aa95
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
examples/FasterRCNN/data.py
View file @ bf683cd2
...@@ -9,7 +9,7 @@ import copy ...@@ -9,7 +9,7 @@ import copy
from tensorpack.utils.argtools import memoized, log_once from tensorpack.utils.argtools import memoized, log_once
from tensorpack.dataflow import ( from tensorpack.dataflow import (
MapData, imgaug, TestDataSpeed, MapData, imgaug, TestDataSpeed,
MapDataComponent, DataFromList, PrefetchDataZMQ) MapDataComponent, DataFromList)
# import tensorpack.utils.viz as tpviz # import tensorpack.utils.viz as tpviz
from coco import COCODetection from coco import COCODetection
...@@ -254,7 +254,7 @@ def get_train_dataflow(add_mask=False): ...@@ -254,7 +254,7 @@ def get_train_dataflow(add_mask=False):
return ret return ret
ds = MapData(ds, preprocess) ds = MapData(ds, preprocess)
ds = PrefetchDataZMQ(ds, 1) # ds = PrefetchDataZMQ(ds, 1)
return ds return ds
...@@ -268,7 +268,7 @@ def get_eval_dataflow(): ...@@ -268,7 +268,7 @@ def get_eval_dataflow():
assert im is not None, fname assert im is not None, fname
return im return im
ds = MapDataComponent(ds, f, 0) ds = MapDataComponent(ds, f, 0)
ds = PrefetchDataZMQ(ds, 1) # ds = PrefetchDataZMQ(ds, 1)
return ds return ds
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment