John F. Wu

icon indicating a website link https://jwuphysics.github.io icon indicating an email [email protected]

icon company Space Telescope Science Institute icon location Baltimore, MD icon location Assistant Astronomer @spacetelescope and Associate Research Scientist at JHU Physics & Astronomy and Computer Science.

Results 3 comments of John F. Wu

getting NaN loss

Unfortunately I'm still getting this error (with `amp=True`) on v0.27.4.

getting NaN loss

I'm using a custom astronomical dataset. I may have set the lr too high (was at 1e-4), but in any event this worked with `amp=False`. Also, I tried to load...

deprecation of pip install sklearn

In `requirements.txt` you'll need to update `sklearn` to `scikit-learn`. Everything else should work with a pip install from the Github repo.