We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7157501 commit 257da0bCopy full SHA for 257da0b
segmentation_models_pytorch/base/utils.py
@@ -1,6 +1,7 @@
1
import torch
2
3
4
+@torch.jit.unused
5
def is_torch_compiling():
6
try:
7
return torch.compiler.is_compiling()
segmentation_models_pytorch/encoders/timm_universal.py
@@ -194,7 +194,7 @@ def output_stride(self) -> int:
194
Returns:
195
int: The effective output stride.
196
"""
197
- return min(self._output_stride, 2**self._depth)
+ return int(min(self._output_stride, 2**self._depth))
198
199
def load_state_dict(self, state_dict, **kwargs):
200
# for compatibility of weights for
0 commit comments