"git@developer.sourcefind.cn:orangecat/ollama.git" did not exist on "d74ce6bd4f78f8a06c97bf9b24485211c48a41d8"
Commit 009a82c2 authored by Ye Liu's avatar Ye Liu Committed by Kai Chen
Browse files

Change 'modelzoo' to 'torchvision' (#1016)

parent 4e387a69
# model settings # model settings
model = dict( model = dict(
type='FasterRCNN', type='FasterRCNN',
pretrained='modelzoo://resnet101', pretrained='torchvision://resnet101',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=101, depth=101,
......
# model settings # model settings
model = dict( model = dict(
type='FasterRCNN', type='FasterRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
...@@ -4,7 +4,7 @@ fp16 = dict(loss_scale=512.) ...@@ -4,7 +4,7 @@ fp16 = dict(loss_scale=512.)
# model settings # model settings
model = dict( model = dict(
type='FasterRCNN', type='FasterRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
...@@ -4,7 +4,7 @@ fp16 = dict(loss_scale=512.) ...@@ -4,7 +4,7 @@ fp16 = dict(loss_scale=512.)
# model settings # model settings
model = dict( model = dict(
type='MaskRCNN', type='MaskRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
...@@ -4,7 +4,7 @@ fp16 = dict(loss_scale=512.) ...@@ -4,7 +4,7 @@ fp16 = dict(loss_scale=512.)
# model settings # model settings
model = dict( model = dict(
type='RetinaNet', type='RetinaNet',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
# model settings # model settings
model = dict( model = dict(
type='MaskRCNN', type='MaskRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
...@@ -3,7 +3,7 @@ norm_cfg = dict(type='SyncBN', requires_grad=True) ...@@ -3,7 +3,7 @@ norm_cfg = dict(type='SyncBN', requires_grad=True)
model = dict( model = dict(
type='MaskRCNN', type='MaskRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
# model settings # model settings
model = dict( model = dict(
type='MaskRCNN', type='MaskRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
...@@ -3,7 +3,7 @@ norm_cfg = dict(type='SyncBN', requires_grad=True) ...@@ -3,7 +3,7 @@ norm_cfg = dict(type='SyncBN', requires_grad=True)
model = dict( model = dict(
type='MaskRCNN', type='MaskRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
...@@ -3,7 +3,7 @@ norm_cfg = dict(type='SyncBN', requires_grad=True) ...@@ -3,7 +3,7 @@ norm_cfg = dict(type='SyncBN', requires_grad=True)
model = dict( model = dict(
type='MaskRCNN', type='MaskRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
# model settings # model settings
model = dict( model = dict(
type='RetinaNet', type='RetinaNet',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
# model settings # model settings
model = dict( model = dict(
type='GridRCNN', type='GridRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
model = dict( model = dict(
type='HybridTaskCascade', type='HybridTaskCascade',
num_stages=3, num_stages=3,
pretrained='modelzoo://resnet101', pretrained='torchvision://resnet101',
interleaved=True, interleaved=True,
mask_info_flow=True, mask_info_flow=True,
backbone=dict( backbone=dict(
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
model = dict( model = dict(
type='HybridTaskCascade', type='HybridTaskCascade',
num_stages=3, num_stages=3,
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
interleaved=True, interleaved=True,
mask_info_flow=True, mask_info_flow=True,
backbone=dict( backbone=dict(
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
model = dict( model = dict(
type='HybridTaskCascade', type='HybridTaskCascade',
num_stages=3, num_stages=3,
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
interleaved=True, interleaved=True,
mask_info_flow=True, mask_info_flow=True,
backbone=dict( backbone=dict(
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
model = dict( model = dict(
type='HybridTaskCascade', type='HybridTaskCascade',
num_stages=3, num_stages=3,
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
interleaved=True, interleaved=True,
mask_info_flow=True, mask_info_flow=True,
backbone=dict( backbone=dict(
......
# model settings # model settings
model = dict( model = dict(
type='FastRCNN', type='FastRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
# model settings # model settings
model = dict( model = dict(
type='FasterRCNN', type='FasterRCNN',
pretrained='modelzoo://resnet101', pretrained='torchvision://resnet101',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=101, depth=101,
......
# model settings # model settings
model = dict( model = dict(
type='FasterRCNN', type='FasterRCNN',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
# model settings # model settings
model = dict( model = dict(
type='RetinaNet', type='RetinaNet',
pretrained='modelzoo://resnet50', pretrained='torchvision://resnet50',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
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