소스 검색

fix: correct formatting in kernel initialization in rec_lcnetv3.py

myhloli 3 주 전
부모
커밋
c922c63fbc
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      mineru/model/utils/pytorchocr/modeling/backbones/rec_lcnetv3.py

+ 1 - 1
mineru/model/utils/pytorchocr/modeling/backbones/rec_lcnetv3.py

@@ -256,7 +256,7 @@ class LearnableRepLayer(nn.Module):
                 input_dim = self.in_channels // self.groups
                 kernel_value = torch.zeros(
                     (self.in_channels, input_dim, self.kernel_size, self.kernel_size),
-                    dtype=branch.weight.dtype,  device= branch.weight.device,
+                    dtype=branch.weight.dtype,  device=branch.weight.device,
                 )
                 for i in range(self.in_channels):
                     kernel_value[