Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

Update TorchScript suffix to *.torchscript (#5856)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-12-02 16:06:45 +01:00
1 parent 30db14fea8
commit 00e308f7be
5 files changed
+20 -20
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

utils/activations.py
+2 -2
View File
@@ -18,8 +18,8 @@ class SiLU(nn.Module): # export-friendly version of nn.SiLU()
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.0, 6.0) / 6.0 # for torchscript, CoreML and ONNX
# return x * F.hardsigmoid(x) # for TorchScript and CoreML
return x * F.hardtanh(x + 3, 0.0, 6.0) / 6.0 # for TorchScript, CoreML and ONNX
# Mish https://github.com/digantamisra98/Mish --------------------------------------------------------------------------