From 1ad0ed1999620c0170e5854dde39805d30d9b6aa Mon Sep 17 00:00:00 2001 From: Vladislav Sovrasov Date: Tue, 14 Apr 2020 16:09:26 +0300 Subject: [PATCH] Update version --- CHANGELOG.md | 4 ++++ setup.py | 5 +---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 52521f5..eb311c0 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,9 @@ # ptflops versions log +## v 0.6 +- Add verbose option to log layers that are not supported by ptflops. +- Add an option to filter a list of operations from the final result. + ## v 0.5.2 - Fix handling of intermediate dimensions in the Linear layer hook. diff --git a/setup.py b/setup.py index d12f33e..bca2f5e 100644 --- a/setup.py +++ b/setup.py @@ -1,11 +1,8 @@ -import os -import shutil -import sys from setuptools import setup, find_packages readme = open('README.md').read() -VERSION = '0.5.2' +VERSION = '0.6' requirements = [ 'torch',