Skip to content

Commit 1f1be17

Browse files
authored
Update requirements.txt (#271)
* Update requirements.txt `timm` version has changed to 0.2.1 two months ago. Is it possible it can change here as well? * Modified Activation Function Input * Update timm_efficientnet.py
1 parent 28744b9 commit 1f1be17

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
torchvision>=0.3.0
22
pretrainedmodels==0.7.4
33
efficientnet-pytorch==0.6.3
4-
timm==0.1.20
4+
timm==0.2.1

segmentation_models_pytorch/encoders/timm_efficientnet.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
11
import torch
22
import torch.nn as nn
33

4-
from timm.models.efficientnet import EfficientNet, Swish
4+
from timm.models.efficientnet import EfficientNet
55
from timm.models.efficientnet import decode_arch_def, round_channels, default_cfgs
6+
from timm.models.layers.activations import Swish
67

78
from ._base import EncoderMixin
89

0 commit comments

Comments
 (0)