Turns out I downloaded the wrong MNIST dataset from somewhere else because the download script is broken. This issue is linked to : Slice layer broken on TensorRT 6.0.1-1
Turns out I downloaded the wrong MNIST dataset from somewhere else because the download script is broken. This issue is linked to : Slice layer broken on TensorRT 6.0.1-1