@ptrblck
I set return crys_fea, out in the forward method, but it gives me an error,
TypeError                                 Traceback (most recent call last)
----> net.fit(SDT_training,target_training)
--->         return super(NeuralNetRegressor, self).fit(X, y, **fit_params)
...
...
TypeError: mse_loss(): argument 'input' (position 1) must be Tensor, not tuple