Skip to content

GitLab

Unverified Commit 6f940375 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Update mask_rcnn_r50_fpn_gn_2x.py

parent 82e75455
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
configs/mask_rcnn_r50_fpn_gn_2x.py
View file @ 6f940375
...@@ -6,7 +6,7 @@ normalize = dict( ...@@ -6,7 +6,7 @@ normalize = dict(
model = dict( model = dict(
type='MaskRCNN', type='MaskRCNN',
pretrained='tools/resnet50-GN.path', pretrained='open-mmlab://contrib/resnet50_gn',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
...@@ -167,14 +167,14 @@ log_config = dict( ...@@ -167,14 +167,14 @@ log_config = dict(
interval=50, interval=50,
hooks=[ hooks=[
dict(type='TextLoggerHook'), dict(type='TextLoggerHook'),
dict(type='TensorboardLoggerHook') # dict(type='TensorboardLoggerHook')
]) ])
# yapf:enable # yapf:enable
# runtime settings # runtime settings
total_epochs = 24 total_epochs = 24
dist_params = dict(backend='nccl') dist_params = dict(backend='nccl')
log_level = 'INFO' log_level = 'INFO'
work_dir = './work_dirs/mask_rcnn_r50_fpn_1x' work_dir = './work_dirs/mask_rcnn_r50_fpn_gn_1x'
load_from = None load_from = None
resume_from = None resume_from = None
workflow = [('train', 1)] workflow = [('train', 1)]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment