@zdevito :  Thanks for the reply. If I had to achieve something similar to freeze_module in PyTorch 1.4.x, is running _jit_pass_lower_graph() and converting the returned parameters into graph constants a good approach? Or is there a better mechanism to achieve this? Porting freeze_module  to 1.4.x seems cumbersome as it relies on AliasDB APIs which don’t exist in 1.4.x.