I used skorch to train my model which is defined as below,
net_reg = NeuralNetRegressor( Network_binaryDC, batch_size=32, lr=0.01, max_epochs=