From 92c83eaaec2c1f37979b36a5e9e7624a637e7832 Mon Sep 17 00:00:00 2001 From: Adrian Bulat Date: Mon, 11 Jan 2021 13:13:46 +0000 Subject: [PATCH] fix versioning, bump version --- README.md | 2 +- pthflops/__init__.py | 2 +- pthflops/ops.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 5d7f3dc..5882ee4 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -[![License](https://img.shields.io/badge/License-BSD%203--Clause-blue.svg)](https://opensource.org/licenses/BSD-3-Clause) [![Build Status](https://travis-ci.com/1adrianb/pytorch-estimate-flops.svg?branch=master)](https://travis-ci.com/1adrianb/pytorch-estimate-flops) +[![License](https://img.shields.io/badge/License-BSD%203--Clause-blue.svg)](https://opensource.org/licenses/BSD-3-Clause) [![Test Pytorch Flops Counter](https://github.com/1adrianb/pytorch-estimate-flops/workflows/Test%20Pytorch%20Flops%20Counter/badge.svg)](https://travis-ci.com/1adrianb/pytorch-estimate-flops) [![PyPI](https://img.shields.io/pypi/v/pthflops.svg?style=flat)](https://pypi.org/project/pthflops/) # pytorch-estimate-flops diff --git a/pthflops/__init__.py b/pthflops/__init__.py index 0f99e89..5787f8f 100644 --- a/pthflops/__init__.py +++ b/pthflops/__init__.py @@ -1,3 +1,3 @@ from .ops import count_ops -__version__ = '0.3.4' +__version__ = '0.3.5' diff --git a/pthflops/ops.py b/pthflops/ops.py index e8b0462..a157935 100644 --- a/pthflops/ops.py +++ b/pthflops/ops.py @@ -309,7 +309,7 @@ def count_ops(model, input, custom_ops={}, ignore_layers=[], print_readable=True if LooseVersion(torch.__version__) >= LooseVersion('1.6.0') and \ LooseVersion(torch.__version__) < LooseVersion('1.8.0'): version = 2 - else: + elif LooseVersion(torch.__version__) >= LooseVersion('1.8.0'): version = 3 else: # PyTorch 1.3 and bellow