Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
seminar-breakout
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Shashank Suhas
seminar-breakout
Commits
5a31fc32
Commit
5a31fc32
authored
Jul 12, 2016
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
separate start_process from prefetch
parent
1095c8b8
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
2 deletions
+4
-2
tensorpack/dataflow/prefetch.py
tensorpack/dataflow/prefetch.py
+4
-2
No files found.
tensorpack/dataflow/prefetch.py
View file @
5a31fc32
...
@@ -125,12 +125,14 @@ class PrefetchDataZMQ(ProxyDataFlow):
...
@@ -125,12 +125,14 @@ class PrefetchDataZMQ(ProxyDataFlow):
self
.
procs
=
[
PrefetchProcessZMQ
(
self
.
ds
,
self
.
pipename
)
self
.
procs
=
[
PrefetchProcessZMQ
(
self
.
ds
,
self
.
pipename
)
for
_
in
range
(
self
.
nr_proc
)]
for
_
in
range
(
self
.
nr_proc
)]
start_proc_mask_signal
(
self
.
procs
)
self
.
start_processes
()
# __del__ not guranteed to get called at exit
# __del__ not guranteed to get called at exit
import
atexit
import
atexit
atexit
.
register
(
lambda
x
:
x
.
__del__
(),
self
)
atexit
.
register
(
lambda
x
:
x
.
__del__
(),
self
)
def
start_processes
(
self
):
start_proc_mask_signal
(
self
.
procs
)
def
get_data
(
self
):
def
get_data
(
self
):
for
k
in
itertools
.
count
():
for
k
in
itertools
.
count
():
if
self
.
_size
>
0
and
k
>=
self
.
_size
:
if
self
.
_size
>
0
and
k
>=
self
.
_size
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment