Quellcode durchsuchen

Update mineru/model/utils/pytorchocr/modeling/backbones/rec_lcnetv3.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Xiaomeng Zhao vor 3 Wochen
Ursprung
Commit
a0da3029fd
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  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[