ai2cm / fv3net

explore the FV3 data for parameterization
MIT License
16 stars 3 forks source link

Novelty detection compatibility with cupy GPU arrays #2399

Closed frodre closed 4 months ago

frodre commented 4 months ago

This PR adds some basic infrastructure for handling Cupy arrays when using novelty detection. Unfortunately due to some of the underlying components, the novelty detection model cannot easily be moved to GPU-only calculations. So for now, we are transferring to CPU where necessary to run the calculation.