Skip to content

Commit

Permalink
[MMSIG open-mmlab#357] Fix missing configs from new cfgs
Browse files Browse the repository at this point in the history
  • Loading branch information
Crescent-Saturn committed Nov 2, 2023
1 parent eb8d80f commit 3cd0652
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 12 deletions.
11 changes: 6 additions & 5 deletions mmdet/configs/panoptic_fpn/panoptic_fpn_r101_fpn_1x_coco.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,9 @@
with read_base():
from .panoptic_fpn_r50_fpn_1x_coco import *

model = dict(
backbone=dict(
depth=101,
init_cfg=dict(
type=PretrainedInit, checkpoint='torchvision://resnet101')))
model.update(
dict(
backbone=dict(
depth=101,
init_cfg=dict(
type=PretrainedInit, checkpoint='torchvision://resnet101'))))
11 changes: 6 additions & 5 deletions mmdet/configs/panoptic_fpn/panoptic_fpn_r101_fpn_ms_3x_coco.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,9 @@
with read_base():
from .panoptic_fpn_r50_fpn_ms_3x_coco import *

model = dict(
backbone=dict(
depth=101,
init_cfg=dict(
type=PretrainedInit, checkpoint='torchvision://resnet101')))
model.update(
dict(
backbone=dict(
depth=101,
init_cfg=dict(
type=PretrainedInit, checkpoint='torchvision://resnet101'))))
4 changes: 2 additions & 2 deletions mmdet/configs/panoptic_fpn/panoptic_fpn_r50_fpn_ms_3x_coco.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,11 +26,11 @@
dict(type=PackDetInputs)
]

train_dataloader = dict(dataset=dict(pipeline=train_pipeline))
train_dataloader.update(dict(dataset=dict(pipeline=train_pipeline)))

# TODO: Use RepeatDataset to speed up training
# training schedule for 3x
train_cfg = dict(max_epochs=36, val_interval=3)
train_cfg.update(dict(max_epochs=36, val_interval=3))

# learning rate
param_scheduler = [
Expand Down

0 comments on commit 3cd0652

Please sign in to comment.