fix flops error

Author: juncaipeng

deploy/python/infer.py

@@ -86,6 +86,7 @@ def __init__(self, args):
             precision_mode = precision_map[args.precision]
 
             if args.use_trt:
+                logger.info("Use TRT")
                 pred_cfg.enable_tensorrt_engine(
                     workspace_size=1 << 30,
                     max_batch_size=1,

paddleseg/models/backbones/resnet_vd.py

@@ -134,7 +134,7 @@ def forward(self, inputs):
         else:
             short = self.short(inputs)
 
-        y = self.add(x=short, y=conv2)
+        y = self.add(short, conv2)
         y = self.relu(y)
         return y
 
@@ -188,7 +188,7 @@ def forward(self, inputs):
             short = inputs
         else:
             short = self.short(inputs)
-        y = self.add(x=short, y=conv1)
+        y = self.add(short, conv1)
         y = self.relu(y)
 
         return y

paddleseg/models/layers/__init__.py

@@ -17,4 +17,4 @@
 from .pyramid_pool import ASPPModule, PPModule
 from .attention import AttentionBlock
 from .nonlocal2d import NonLocal2D
-from .wrap_function import *
+from .wrap_functions import *