Commit 77cd0481 authored by Daniel Williams's avatar Daniel Williams 🤖
Browse files

Started adding documentation for the new gpytorch models.

parent d3f9c454
GPyTorch-based models
A number of models implemented in ``heron`` make use of ``pytorch`` and the GPR library built atop it, ``gpytorch``.
These models can be used on both CPU and GPU hardware.
All of these models are contained within the `heron.models.torchbased` module.
HeronCUDA : A spinning, NR-trained, GPU-capable surrogate model
| Training data | GPR Technique | Model type | Spinning | Higher modes |
| NR: Georgia Tech | Exact, LOVE, CUDA | BBH | Fully | No |
The model is trained on `numerical relativity waveforms <>`_ produced by the Centre for Relativistic Astrophysics at Georgia Tech, and uses exact scalable GPR techniques implemented by `GPyTorch <>`_.
.. autoclass:: heron.models.torchbased.HeronCUDA
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment