Merge pull request tusen-ai#35 from joyhuang9473/fix-hard-coding-with-num-of-branch-setting

Fix hard coding with num_branch setting

Author: huangzehao

models/tridentnet/resnet_v2_for_paper.py

@@ -200,7 +200,7 @@ def resnet_trident_stage(cls, data, name, num_block, filter, stride, dilate, nor
                 unit_deform = False
             # cast back to fp32 as deformable conv is not optimized for fp16
             if unit_deform and i == num_block - 2:
-                for j in range(3):
+                for j in range(num_branch):
                     data[j] = X.to_fp32(data[j], name="deform_to32")
             data = cls.resnet_trident_unit(
                 data, "{}_unit{}".format(name, i), filter, (1, 1), d, False, norm_type, norm_mom, ndev,