Skip to content

A
ai-stock-analysis
Switch branch/tag
History Find file
  • xeron56's avatar
    Add CUDA device compatibility checks to TimesFM backend · 6132f5c0
    xeron56 authored 
    - Implemented a function to check if the current CUDA device is supported by the installed PyTorch version.
- Added warnings for unsupported CUDA devices, informing users that the CPU will be used instead.
- Updated the TimesFMBackend class to utilize the new compatibility checks and handle GPU names appropriately.
- Introduced unit tests to validate the behavior of the CUDA compatibility function with various simulated CUDA environments.
    6132f5c0
Name
Last commit
 Last update
..
app.component.html Loading commit data...
app.component.scss Loading commit data...
app.component.ts Loading commit data...
stock-analysis.model.ts Loading commit data...
stock-analysis.service.ts Loading commit data...
timesfm-prediction.component.html Loading commit data...
timesfm-prediction.component.scss Loading commit data...
timesfm-prediction.component.ts Loading commit data...
timesfm-prediction.model.ts Loading commit data...
timesfm-prediction.service.ts Loading commit data...