Merge pull request pytorch#10 from o295/main

xiaohu2015 · web-flow · commit 9fada5655fb8 · 2021-11-26T14:21:15.000+08:00
Fixing test_detection_model_trainable_backbone_layers
diff --git a/test/test_models.py b/test/test_models.py
@@ -321,6 +321,10 @@ def _check_input_backprop(model, inputs):
         "max_trainable": 6,
         "n_trn_params_per_layer": [96, 99, 138, 200, 239, 257, 266],
     },
+    "fcos_resnet50_fpn": {
+        "max_trainable": 5,
+        "n_trn_params_per_layer": [54, 64, 83, 96, 106, 107],
+    },
 }
 
 

Original file line number	Diff line number	Diff line change
`@@ -321,6 +321,10 @@ def _check_input_backprop(model, inputs):`
`321`	`321`	`"max_trainable": 6,`
`322`	`322`	`"n_trn_params_per_layer": [96, 99, 138, 200, 239, 257, 266],`
`323`	`323`	`},`
	`324`	`+ "fcos_resnet50_fpn": {`
	`325`	`+ "max_trainable": 5,`
	`326`	`+ "n_trn_params_per_layer": [54, 64, 83, 96, 106, 107],`
	`327`	`+ },`
`324`	`328`	`}`
`325`	`329`
`326`	`330`