Spaces:
Runtime error
Runtime error
Upload resnet50.py
Browse files
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 |
+
))
|