Ok maybe you want to increase the Tensor sizes A LOT 
Thinking back at it, they are very very small.
Maybe do something like a regular image: 1, 3, 240, 240 for the medium and something even bigger for the large