update CHANGELOG.md

Author: lezwon

CHANGELOG.md

@@ -1,3 +1,4 @@
+
 # Changelog
 
 All notable changes to this project will be documented in this file.
@@ -8,7 +9,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 ## [1.1.4] - YYYY-MM-DD
 
 ### Added
-
+- Support to tie weights after moving model to TPU via `on_post_move_to_device` hook
 
 ### Changed
 

pytorch_lightning/core/decorators.py

@@ -76,8 +76,6 @@ def inner_f(self, *args, **kwargs):
         post_param_count = len(list(self.parameters()))
 
         if not pre_param_count == post_param_count:
-            import os
-            print(os.environ.get('LOCAL_RANK'))
             rank_zero_warn('The model parameters do not match after moving to the target device. '
                            'If your model employs weight sharing on TPU,'
                            'please tie your weights using the `on_post_move_to_device` model hook.')