While designing a Neural Network, we can use Dataset & Data Loader for efficient memory management. Think of them as a backbone for providing data to a machine learning model. Optuna package available in the PyTorch library will be used to calculate the hyper parameters. Agenda of this blog : How we are making the model memory efficient by using Dataset, Data Loader How to use Optuna package to select accurate Hyper Parameters Hyper parameters in a Neural Network : Learning rate Batch size No. of epochs Drop out ratio No. of hidden layers No. of hidden units Normalization - Batch norm etc. Optimizers - Adam, Momentum, NAG, RMS Prop etc. Generally, we manually try above hyper parameters with different reasonable values and try testing a model and finalize the accurate paramters which are best suitable for our model. But manual way takes time as we need to see different possibilities by using different values for these parameters and then come up with correct parameters and values....