I am creating a neural network for PA behavioral modeling. The model takes in input I and Q values and predicts the amplifier output. So that’s a .csv file with two columns. One of my goals is to prune (or optimize in any way) the model I built. The original model works just fine. However, I am facing problems while pruning the created, trained, and tested model.