Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Force features to be floats instead of doubles
As per @caitlynlee in google-research#7 , Make predictions give errors for various tensors as `expected dtype double does not equal original dtype float`. Forcing tensors to be floats instead of doubles in `astronet/predict.py ` solves the issue. ``` 115: global_view = preprocess.global_view(time, flux, FLAGS.period).astype(np.float32) ... 120: local_view = preprocess.local_view(time, flux, FLAGS.period, FLAGS.duration).astype(np.float32) ```
- Loading branch information