From c670af21721c103c4d3816654acb561eb49f5b5e Mon Sep 17 00:00:00 2001
From: kelisiya <402124496@qq.com>
Date: Thu, 27 May 2021 14:09:30 +0800
Subject: [PATCH] Update repvgg.py

conv_bn() got an unexpected keyword argument 'dilation'
---
 repvgg.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/repvgg.py b/repvgg.py
index d9d11bc..385c330 100644
--- a/repvgg.py
+++ b/repvgg.py
@@ -38,7 +38,7 @@ def __init__(self, in_channels, out_channels, kernel_size,
 
         else:
             self.rbr_identity = nn.BatchNorm2d(num_features=in_channels) if out_channels == in_channels and stride == 1 else None
-            self.rbr_dense = conv_bn(in_channels=in_channels, out_channels=out_channels, kernel_size=kernel_size, stride=stride, padding=padding, dilation=dilation, groups=groups)
+            self.rbr_dense = conv_bn(in_channels=in_channels, out_channels=out_channels, kernel_size=kernel_size, stride=stride, padding=padding,  groups=groups)
             self.rbr_1x1 = conv_bn(in_channels=in_channels, out_channels=out_channels, kernel_size=1, stride=stride, padding=padding_11, groups=groups)
             print('RepVGG Block, identity = ', self.rbr_identity)