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
fc0f6fca
Commit
fc0f6fca
authored
Jan 25, 2018
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
some docs update
parent
b2a396e5
Changes
5
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
16 additions
and
9 deletions
+16
-9
.github/ISSUE_TEMPLATE.md
.github/ISSUE_TEMPLATE.md
+1
-1
examples/FasterRCNN/config.py
examples/FasterRCNN/config.py
+1
-0
examples/FasterRCNN/train.py
examples/FasterRCNN/train.py
+5
-1
tensorpack/train/base.py
tensorpack/train/base.py
+2
-2
tensorpack/train/config.py
tensorpack/train/config.py
+7
-5
No files found.
.github/ISSUE_TEMPLATE.md
View file @
fc0f6fca
Bug Reports/Feature Requests/Usage Questions Only:
Any unexpected problems
with code
: PLEASE always include
Any unexpected problems: PLEASE always include
1.
What you did. (command you run and changes you made if using examples; post or describe your code if not)
2.
What you observed, e.g. as much as logs possible.
3.
What you expected, if not obvious.
...
...
examples/FasterRCNN/config.py
View file @
fc0f6fca
...
...
@@ -65,4 +65,5 @@ TEST_PRE_NMS_TOPK = 6000
TEST_POST_NMS_TOPK
=
1000
# if you encounter OOM in inference, set this to a smaller number
FASTRCNN_NMS_THRESH
=
0.5
RESULT_SCORE_THRESH
=
0.05
RESULT_SCORE_THRESH_VIS
=
0.3
# only visualize confident results
RESULTS_PER_IM
=
100
examples/FasterRCNN/train.py
View file @
fc0f6fca
...
...
@@ -343,6 +343,10 @@ if __name__ == '__main__':
assert
args
.
load
print_config
()
if
args
.
predict
or
args
.
visualize
:
config
.
RESULT_SCORE_THRESH
=
config
.
RESULT_SCORE_THRESH_VIS
if
args
.
visualize
:
visualize
(
args
.
load
)
else
:
...
...
@@ -355,7 +359,7 @@ if __name__ == '__main__':
assert
args
.
evaluate
.
endswith
(
'.json'
)
offline_evaluate
(
pred
,
args
.
evaluate
)
elif
args
.
predict
:
COCODetection
(
config
.
BASEDIR
,
'
train2014'
)
#
to load the class names into caches
COCODetection
(
config
.
BASEDIR
,
'
val2014'
)
# Only
to load the class names into caches
predict
(
pred
,
args
.
predict
)
else
:
logger
.
set_logger_dir
(
args
.
logdir
)
...
...
tensorpack/train/base.py
View file @
fc0f6fca
...
...
@@ -294,8 +294,8 @@ class Trainer(object):
"""
Same as :meth:`train()`, but will:
1. Append `DEFAULT_CALLBACKS()` to callbacks.
2. Append `DEFAULT_MONITORS()` to monitors.
1. Append
:meth:
`DEFAULT_CALLBACKS()` to callbacks.
2. Append
:meth:
`DEFAULT_MONITORS()` to monitors.
3. Provide default values for every option except `steps_per_epoch`.
"""
callbacks
=
(
callbacks
or
[])
+
DEFAULT_CALLBACKS
()
...
...
tensorpack/train/config.py
View file @
fc0f6fca
...
...
@@ -67,12 +67,14 @@ class TrainConfig(object):
callbacks (list): a list of :class:`Callback` to perform during training.
extra_callbacks (list): the same as ``callbacks``. This argument
is only used to provide the defaults in addition to ``callbacks``. The defaults are
``MovingAverageSummary()``, ``ProgressBar()``,
``MergeAllSummaries()``, ``RunUpdateOps()``. The list of
callbacks that will be used in the end is ``callbacks + extra_callbacks``.
is only used to provide the defaults in addition to ``callbacks``.
The list of callbacks that will be used in the end is ``callbacks + extra_callbacks``.
It is usually left as None and the default value for this
option will be the return value of :meth:`train.DEFAULT_CALLBACKS()`.
You can override it when you don't like any of the default callbacks.
monitors (list): a list of :class:`TrainingMonitor`.
Defaults to
``TFEventWriter()``, ``JSONWriter()``, ``ScalarPrinter()`
`.
Defaults to
the return value of :meth:`train.DEFAULT_MONITORS()
`.
session_creator (tf.train.SessionCreator): Defaults to :class:`sesscreate.NewSessionCreator()`
with the config returned by :func:`tfutils.get_default_sess_config()`.
...
...
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