Hyperparameter optimization in pytorch (currently with sklearn GridSearchCV)

Question

I use this(link) pytorch tutorial and wish to add the grid search functionality in it ,sklearn.model_selection.GridSearchCV (link), in order to optimize the hyper parameters