Convert Sklearn MLPRegressor to Pytorch

Hello, I am trying converting my MLP regression model to pytorch. This is my first time using Pytorch. below is the simple MLP model:
reg=MLPRegressor()
reg.fit(x_train, y_train)
pred=reg.predict(x_test)
Can someone help me to figure out what changes I need to make in order to convert this sklearn model to Pytorch model? Tahnks in advance for your help.

This post might be useful, as it’s dealing with the same problem. :slight_smile: