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
e53bf227
Commit
e53bf227
authored
Oct 01, 2019
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Make disable_layer_logging public
parent
cc2322bb
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
6 additions
and
5 deletions
+6
-5
tensorpack/models/common.py
tensorpack/models/common.py
+3
-2
tensorpack/models/registry.py
tensorpack/models/registry.py
+1
-1
tensorpack/utils/concurrency.py
tensorpack/utils/concurrency.py
+2
-2
No files found.
tensorpack/models/common.py
View file @
e53bf227
# -*- coding: utf-8 -*-
# -*- coding: utf-8 -*-
# File: common.py
# File: common.py
from
.registry
import
layer_register
# noqa
from
.registry
import
layer_register
,
disable_layer_logging
# noqa
from
.tflayer
import
rename_tflayer_get_variable
from
.tflayer
import
rename_tflayer_get_variable
from
.utils
import
VariableHolder
# noqa
from
.utils
import
VariableHolder
# noqa
__all__
=
[
'layer_register'
,
'VariableHolder'
,
'rename_tflayer_get_variable'
]
__all__
=
[
'layer_register'
,
'VariableHolder'
,
'rename_tflayer_get_variable'
,
'disable_layer_logging'
]
tensorpack/models/registry.py
View file @
e53bf227
...
@@ -18,7 +18,7 @@ from ..utils import logger
...
@@ -18,7 +18,7 @@ from ..utils import logger
_LAYER_LOGGED
=
set
()
_LAYER_LOGGED
=
set
()
_LAYER_REGISTRY
=
{}
_LAYER_REGISTRY
=
{}
__all__
=
[
'layer_register'
]
__all__
=
[
'layer_register'
,
'disable_layer_logging'
]
_NameConflict
=
"LAYER_NAME_CONFLICT!!"
_NameConflict
=
"LAYER_NAME_CONFLICT!!"
...
...
tensorpack/utils/concurrency.py
View file @
e53bf227
...
@@ -237,8 +237,8 @@ def start_proc_mask_signal(proc):
...
@@ -237,8 +237,8 @@ def start_proc_mask_signal(proc):
if
sys
.
version_info
<
(
3
,
4
)
or
mp
.
get_start_method
()
==
'fork'
:
if
sys
.
version_info
<
(
3
,
4
)
or
mp
.
get_start_method
()
==
'fork'
:
log_once
(
log_once
(
"Starting a process with 'fork' method is not safe and may consume unnecessary extra CPU 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. "
" Use 'forkserver
' or '
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"
,
"See https://docs.python.org/3/library/multiprocessing.html#contexts-and-start-methods
on how to set them.
"
,
'warn'
)
# noqa
'warn'
)
# noqa
p
.
start
()
p
.
start
()
...
...
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