Filter "unsqueeze" warning when using DP (#5622)
Filter "unsqueeze" warning when using DP (#5622)

* filtering unsqueeze warning directly in DP

* add changelog

* constrain to module

Co-authored-by: mergify[bot] <b09a6ee808b67e98a221404e7aaa52e0398a4954@users.noreply.github.com>
Co-authored-by: Nicki Skafte <6834dde87a397b36fec9709e4a16ddb6dba8b76b@gmail.com>
Co-authored-by: Jirka Borovec <b903e28b3fb12ead55c25f3a7816376be3bd063c@users.noreply.github.com>

williamFalcon/pytorch-lightning,williamFalcon/pytorch-lightning