Make sure that the predictions from the trainer / PyTorch match the predictions from NeuralAmpModelerCore.
Make sure that the predictions from the trainer / PyTorch match the predictions from NeuralAmpModelerCore.