From a56c4de4804f31f6c3b95b335b36a37279b69f47 Mon Sep 17 00:00:00 2001 From: yihui-he Date: Sat, 1 Apr 2023 11:22:18 -0700 Subject: [PATCH] add channel-pruning MIT license Signed-off-by: yihui-he --- Docs/user_guide/channel_pruning.rst | 1 + .../python/aimet_tensorflow/channel_pruning/channel_pruner.py | 4 ++++ .../src/python/aimet_torch/channel_pruning/channel_pruner.py | 4 ++++ 3 files changed, 9 insertions(+) diff --git a/Docs/user_guide/channel_pruning.rst b/Docs/user_guide/channel_pruning.rst index 6b3a0b812d6..727fed63e38 100644 --- a/Docs/user_guide/channel_pruning.rst +++ b/Docs/user_guide/channel_pruning.rst @@ -4,6 +4,7 @@ AIMET Channel Pruning Channel Pruning is a model compression technique that reduces less-important input channels from layers in a given model. Currently AIMET supports Channel Pruning of Conv2d layers. +For more background details about Channel Pruning, please refer to He, Yihui, Xiangyu Zhang, and Jian Sun. Channel pruning for accelerating very deep neural networks. Overall Procedure diff --git a/TrainingExtensions/tensorflow/src/python/aimet_tensorflow/channel_pruning/channel_pruner.py b/TrainingExtensions/tensorflow/src/python/aimet_tensorflow/channel_pruning/channel_pruner.py index ec28304a311..6addeb32fc4 100644 --- a/TrainingExtensions/tensorflow/src/python/aimet_tensorflow/channel_pruning/channel_pruner.py +++ b/TrainingExtensions/tensorflow/src/python/aimet_tensorflow/channel_pruning/channel_pruner.py @@ -35,6 +35,10 @@ # # @@-COPYRIGHT-END-@@ # ============================================================================= +# +# This file contains code adapted from channel-pruning (https://github.com/yihui-he/channel-pruning) +# Copyright (c) 2018 Yihui He 何宜晖 +# Licensed under the MIT License (see https://github.com/yihui-he/channel-pruning/blob/master/LICENSE for details) """ Prunes layers using Channel Pruning scheme """ diff --git a/TrainingExtensions/torch/src/python/aimet_torch/channel_pruning/channel_pruner.py b/TrainingExtensions/torch/src/python/aimet_torch/channel_pruning/channel_pruner.py index be8504d18f7..95a4bb20bda 100644 --- a/TrainingExtensions/torch/src/python/aimet_torch/channel_pruning/channel_pruner.py +++ b/TrainingExtensions/torch/src/python/aimet_torch/channel_pruning/channel_pruner.py @@ -35,6 +35,10 @@ # # @@-COPYRIGHT-END-@@ # ============================================================================= +# +# This file contains code adapted from channel-pruning (https://github.com/yihui-he/channel-pruning) +# Copyright (c) 2018 Yihui He 何宜晖 +# Licensed under the MIT License (see https://github.com/yihui-he/channel-pruning/blob/master/LICENSE for details) """ Prunes layers using Channel Pruning scheme """ from typing import Iterator, List