Fix float zeros format (#5491)

* Fix float zeros format

* 255 to integer
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-11-03 23:36:53 +01:00
1 parent 8a803f36d3
commit 5866646cc8
16 files changed
+36 -36

No files matched your search

+1 -1
View File
@@ -19,7 +19,7 @@ class Hardswish(nn.Module): # export-friendly version of nn.Hardswish()
@staticmethod
def forward(x):
# return x * F.hardsigmoid(x) # for torchscript and CoreML
return x * F.hardtanh(x + 3, 0., 6.) / 6. # for torchscript, CoreML and ONNX
return x * F.hardtanh(x + 3, 0.0, 6.0) / 6.0 # for torchscript, CoreML and ONNX
# Mish https://github.com/digantamisra98/Mish --------------------------------------------------------------------------