KyanChen commited on
Commit
75edc1d
1 Parent(s): 2fed23f

Upload resnet50.py

Browse files
Files changed (1) hide show
  1. configs/_base_/models/resnet50.py +17 -0
configs/_base_/models/resnet50.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # model settings
2
+ model = dict(
3
+ type='ImageClassifier',
4
+ backbone=dict(
5
+ type='ResNet',
6
+ depth=50,
7
+ num_stages=4,
8
+ out_indices=(3, ),
9
+ style='pytorch'),
10
+ neck=dict(type='GlobalAveragePooling'),
11
+ head=dict(
12
+ type='LinearClsHead',
13
+ num_classes=1000,
14
+ in_channels=2048,
15
+ loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
16
+ topk=(1, 5),
17
+ ))