Gradient Explosion #2413
Description
Running tests/estimators/regression/test_pytorch_regression.py (specifically, setUpClass) causes large gradient values in this line.
{{ message }}
Running tests/estimators/regression/test_pytorch_regression.py (specifically, setUpClass) causes large gradient values in this line.