diff --git a/learners/channel_pruning/learner.py b/learners/channel_pruning/learner.py
index 689e9ac..28a35b2 100644
--- a/learners/channel_pruning/learner.py
+++ b/learners/channel_pruning/learner.py
@@ -14,6 +14,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
+# 
+# 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)
+
 """Channel Pruned Learner"""
 import os
 import math