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
860efcf0
Commit
860efcf0
authored
Nov 16, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[FasterRCNN] using the wrong schedule for the published result
parent
17261566
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
3 deletions
+3
-3
examples/FasterRCNN/README.md
examples/FasterRCNN/README.md
+1
-1
examples/FasterRCNN/train.py
examples/FasterRCNN/train.py
+2
-2
No files found.
examples/FasterRCNN/README.md
View file @
860efcf0
...
...
@@ -47,7 +47,7 @@ To evaluate the performance (pretrained models can be downloaded in [model zoo](
Mean Average Precision @IoU=0.50:0.95:
+
trainval35k/minival, FASTRCNN_BATCH=256: 34.2. Takes 49h on 8 TitanX.
+
trainval35k/minival, FASTRCNN_BATCH=64: 32.7. Takes
25
h on 8 TitanX.
+
trainval35k/minival, FASTRCNN_BATCH=64: 32.7. Takes
31
h on 8 TitanX.
The hyperparameters are not carefully tuned. You can probably get better performance by e.g. training longer.
...
...
examples/FasterRCNN/train.py
View file @
860efcf0
...
...
@@ -320,12 +320,12 @@ if __name__ == '__main__':
'learning_rate'
,
[(
warmup_epoch
*
factor
,
1e-2
),
(
150000
*
factor
//
stepnum
,
1e-3
),
(
2
1
0000
*
factor
//
stepnum
,
1e-4
)]),
(
2
3
0000
*
factor
//
stepnum
,
1e-4
)]),
EvalCallback
(),
GPUUtilizationTracker
(),
],
steps_per_epoch
=
stepnum
,
max_epoch
=
2
3
0000
*
factor
//
stepnum
,
max_epoch
=
2
8
0000
*
factor
//
stepnum
,
session_init
=
get_model_loader
(
args
.
load
)
if
args
.
load
else
None
,
)
trainer
=
SyncMultiGPUTrainerReplicated
(
get_nr_gpu
())
...
...
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