Skip to content

Commit 2a7d256

Browse files
committed
Re-enable mem-efficient/jit activations after torchscript tests
1 parent f902bcd commit 2a7d256

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/layers/activations.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from torch.nn import functional as F
1313

1414

15-
_USE_MEM_EFFICIENT_ISH = False
15+
_USE_MEM_EFFICIENT_ISH = True
1616
if _USE_MEM_EFFICIENT_ISH:
1717
# This version reduces memory overhead of Swish during training by
1818
# recomputing torch.sigmoid(x) in backward instead of saving it.

0 commit comments

Comments
 (0)