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
140778d4
Commit
140778d4
authored
Jul 31, 2019
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
update docs
parent
c9fde630
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
2 deletions
+4
-2
tensorpack/train/trainers.py
tensorpack/train/trainers.py
+3
-1
tensorpack/utils/concurrency.py
tensorpack/utils/concurrency.py
+1
-1
No files found.
tensorpack/train/trainers.py
View file @
140778d4
...
...
@@ -226,6 +226,7 @@ class DistributedTrainerBase(SingleCostTrainer):
get_distributed_session_creator
(
self
.
server
),
session_init
)
# This is slow. deprecated in favor of horovod
class
DistributedTrainerParameterServer
(
DistributedTrainerBase
):
__doc__
=
DistributedParameterServerBuilder
.
__doc__
...
...
@@ -253,6 +254,7 @@ class DistributedTrainerParameterServer(DistributedTrainerBase):
return
[]
# This is slow. deprecated in favor of horovod
class
DistributedTrainerReplicated
(
DistributedTrainerBase
):
__doc__
=
DistributedReplicatedBuilder
.
__doc__
...
...
@@ -275,7 +277,7 @@ class DistributedTrainerReplicated(DistributedTrainerBase):
def
_setup_input
(
self
,
input_signature
,
input
):
with
override_to_local_variable
():
get_global_step_var
()
# gs should be local
# input source may create variable (queue size summary)
# input source may create variable
s
(queue size summary)
# TODO This is not good because we don't know from here
# whether something should be global or local. We now assume
# they should be local.
...
...
tensorpack/utils/concurrency.py
View file @
140778d4
...
...
@@ -237,7 +237,7 @@ def start_proc_mask_signal(proc):
if
isinstance
(
p
,
mp
.
Process
):
if
sys
.
version_info
<
(
3
,
4
)
or
mp
.
get_start_method
()
==
'fork'
:
log_once
(
"Starting a process with 'fork' method is not safe and may consume unnecessary extra memory."
"Starting a process with 'fork' method is not safe and may consume unnecessary extra
CPU
memory."
" Use 'forkserver/spawn' method (available after Py3.4) instead if you run into any issues. "
"See https://docs.python.org/3/library/multiprocessing.html#contexts-and-start-methods"
,
'warn'
)
# noqa
...
...
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