Suburst commited on
Commit
34d4bd0
·
verified ·
1 Parent(s): 556e597

Update Yolov5_Deepsort/models/common.py

Browse files
Files changed (1) hide show
  1. Yolov5_Deepsort/models/common.py +1 -0
Yolov5_Deepsort/models/common.py CHANGED
@@ -39,6 +39,7 @@ class Conv(nn.Module):
39
  self.act = nn.SiLU() if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
40
 
41
  def forward(self, x):
 
42
  return self.act(self.bn(self.conv(x)))
43
 
44
  def fuseforward(self, x):
 
39
  self.act = nn.SiLU() if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
40
 
41
  def forward(self, x):
42
+ x = x.float()
43
  return self.act(self.bn(self.conv(x)))
44
 
45
  def fuseforward(self, x):