search

logix-project / logix

AI Logging for Interpretability and Explainability🔬
Apache License 2.0
74 stars 6 forks source link

Configurable CPU Offloading #81

Closed sangkeun00 closed 8 months ago

sangkeun00 commented 8 months ago

Depending on users' setup, they may afford keeping all statistic states (e.g. mean, covariance) on GPU. Instead of enforcing users to always use CPU offloading, we can let them configure this in config.yaml as follows:

logging:
    cpu_offload: true/false
sangkeun00 commented 8 months ago

Addressed in #82