[style] rename RepConv to RepVGGBlock

Nota-NetsPresso · Nov 14, 2024 · da7bc40 · da7bc40
1 parent 10931a9
commit da7bc40
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/netspresso_trainer/models/necks/experimental/rtdetr_hybrid_encoder.py b/src/netspresso_trainer/models/necks/experimental/rtdetr_hybrid_encoder.py
@@ -28,7 +28,7 @@
 
 from ...utils import BackboneOutput
 from ...op.registry import ACTIVATION_REGISTRY
-from ...op.custom import RepConv, ConvLayer, CSPRepLayer
+from ...op.custom import RepVGGBlock, ConvLayer, CSPRepLayer
 
 
 # transformer

diff --git a/src/netspresso_trainer/models/op/custom.py b/src/netspresso_trainer/models/op/custom.py
@@ -829,7 +829,7 @@ def __init__(self,
         self.conv1 = ConvLayer(in_channels, hidden_channels, kernel_size=1, stride=1, bias=bias, act_type=act)
         self.conv2 = ConvLayer(in_channels, hidden_channels, kernel_size=1, stride=1, bias=bias, act_type=act)
         self.bottlenecks = nn.Sequential(*[
-            RepConv(hidden_channels, hidden_channels, act_type=act) for _ in range(num_blocks)
+            RepVGGBlock(hidden_channels, hidden_channels, act_type=act) for _ in range(num_blocks)
         ])
         if hidden_channels != out_channels:
             self.conv3 = ConvLayer(hidden_channels, out_channels, kernel_size=1, stride=1, bias=bias, act_type=act)