MTLTextureType2D, expected MTLTextureType2DArray

[bocsiboti]

I followed the listed steps to train on the MNIST dataset. However, after starting the training, I get the following error:

validateComputeFunctionArguments:816: failed assertion `Compute Function(MPSCNNLoss): incorrect type of texture (MTLTextureType2D) bound at texture binding at index 2 (expect MTLTextureType2DArray) for dst_2DArray[0].'

in this line (TrainingViewController:797):

let _ = graph!.encodeBatch(to: commandBuffer, sourceImages: [trainingImages], sourceStates: [lossLabels])

I am using: macOS version 10.15.3 XCode Version 11.3.1 (11C504) SwiftProtobuf 1.8.0