-
Notifications
You must be signed in to change notification settings - Fork 0
Open
Labels
featureadd new featadd new feat
Description
Currently, GPU usage is not explicitly handled in the model base class.
To provide better flexibility, we should introduce a mechanism to switch between NumPy and CuPy based on user preference.
Proposed Solution:
- Convert
npinto a class attribute in the model base class. - If the user wants to use the GPU, set
npto CuPy (cupy). - Otherwise, default to NumPy (
numpy).
This change will allow seamless switching between CPU and GPU computations without modifying other parts of the code.
Additional Context:
This approach will improve performance for users who want GPU acceleration while maintaining compatibility for CPU-only environments.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
featureadd new featadd new feat