ClownRat commited on
Commit
b4b364b
·
verified ·
1 Parent(s): f91ca21

modify output indices

Browse files
Files changed (1) hide show
  1. config.json +6 -0
config.json CHANGED
@@ -2025,9 +2025,15 @@
2025
  "model_type": "resnet",
2026
  "num_channels": 3,
2027
  "out_features": [
 
 
 
2028
  "stage4"
2029
  ],
2030
  "out_indices": [
 
 
 
2031
  4
2032
  ],
2033
  "stage_names": [
 
2025
  "model_type": "resnet",
2026
  "num_channels": 3,
2027
  "out_features": [
2028
+ "stage1",
2029
+ "stage2",
2030
+ "stage3",
2031
  "stage4"
2032
  ],
2033
  "out_indices": [
2034
+ 1,
2035
+ 2,
2036
+ 3,
2037
  4
2038
  ],
2039
  "stage_names": [