Skip to content

Commit

Permalink
Add reference of the RepVGG block
Browse files Browse the repository at this point in the history
  • Loading branch information
hglee98 committed Nov 14, 2024
1 parent 6a9bbfd commit 10931a9
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/netspresso_trainer/models/op/custom.py
Original file line number Diff line number Diff line change
Expand Up @@ -175,6 +175,7 @@ def forward(self, x: Union[Tensor, Proxy]) -> Union[Tensor, Proxy]:
class RepVGGBlock(nn.Module):
"""
A convolutional block that combines two convolution layers (kernel and point-wise conv).
This implementation is based on https://github.com/lyuwenyu/RT-DETR/blob/b444daf79cf25f95b740ae71e80fd165e892739a/rtdetr_pytorch/src/zoo/rtdetr/hybrid_encoder.py#L35.
"""
def __init__(self,
in_channels: int,
Expand Down

0 comments on commit 10931a9

Please sign in to comment.