Преглед на файлове

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[