GitLab

Add in the reshape parameter to better mirror what we're seeing in the retinanet
network block that's causing the issue