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
339f2173
Commit
339f2173
authored
Nov 29, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[FasterRCNN] always save the last model
parent
bedec8cd
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
examples/FasterRCNN/train.py
examples/FasterRCNN/train.py
+1
-1
tensorpack/callbacks/saver.py
tensorpack/callbacks/saver.py
+1
-1
No files found.
examples/FasterRCNN/train.py
View file @
339f2173
...
@@ -365,7 +365,7 @@ if __name__ == '__main__':
...
@@ -365,7 +365,7 @@ if __name__ == '__main__':
model
=
Model
(),
model
=
Model
(),
data
=
QueueInput
(
get_train_dataflow
(
add_mask
=
config
.
MODE_MASK
)),
data
=
QueueInput
(
get_train_dataflow
(
add_mask
=
config
.
MODE_MASK
)),
callbacks
=
[
callbacks
=
[
PeriodicTrigger
(
ModelSaver
(),
every_k_epochs
=
5
),
ModelSaver
(
max_to_keep
=
10
,
keep_checkpoint_every_n_hours
=
1
),
# linear warmup
# linear warmup
ScheduledHyperParamSetter
(
ScheduledHyperParamSetter
(
'learning_rate'
,
'learning_rate'
,
...
...
tensorpack/callbacks/saver.py
View file @
339f2173
...
@@ -25,7 +25,7 @@ class ModelSaver(Callback):
...
@@ -25,7 +25,7 @@ class ModelSaver(Callback):
"""
"""
Args:
Args:
max_to_keep (int): the same as in ``tf.train.Saver``.
max_to_keep (int): the same as in ``tf.train.Saver``.
keep_checkpoint_every_n_hours (
in
t): the same as in ``tf.train.Saver``.
keep_checkpoint_every_n_hours (
floa
t): the same as in ``tf.train.Saver``.
checkpoint_dir (str): Defaults to ``logger.get_logger_dir()``.
checkpoint_dir (str): Defaults to ``logger.get_logger_dir()``.
var_collections (str or list of str): collection of the variables (or list of collections) to save.
var_collections (str or list of str): collection of the variables (or list of collections) to save.
"""
"""
...
...
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