Energy disaggregation (Pytorch implementation)