diff --git a/requirements.txt b/requirements.txt index cefb3ef6..41b747ac 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ torchvision>=0.3.0 pretrainedmodels==0.7.4 -efficientnet-pytorch>=0.6.3 +efficientnet-pytorch==0.6.3 timm==0.1.20 diff --git a/segmentation_models_pytorch/__version__.py b/segmentation_models_pytorch/__version__.py index 1f1760fa..de812640 100644 --- a/segmentation_models_pytorch/__version__.py +++ b/segmentation_models_pytorch/__version__.py @@ -1,3 +1,3 @@ -VERSION = (0, 1, 1) +VERSION = (0, 1, 2) __version__ = '.'.join(map(str, VERSION))