Unable to train | tensorflow.python.framework.errors_impl.InvalidArgumentError: Conv2DCustomBackpropInputOp only supports NHWC

The gpu is a bit old. Please refer to Training on Custom Dataset using TLT