This repository has been archived by the owner on Oct 9, 2024. It is now read-only.

Add accelerator device arguments and cuda test #82

Merged

martinkim0 merged 2 commits into main from devices

Jan 31, 2024

Member

martinkim0 commented Jan 31, 2024 •

edited

Loading

Adds back accelerator and devices arguments
Adds specification for running tests on CUDA
Bumps scvi-tools lower bound to 1.1.0rc2 due to an Optax patch


          Add accelerator device arguments and cuda test

84176ff

martinkim0 added the cuda tests label


          Bump scvi req to 1.1.0rc2

b19e069

martinkim0 added all tests and removed cuda tests labels

martinkim0 merged commit 42612ff into main

9 of 10 checks passed

martinkim0 deleted the devices branch

January 31, 2024 18:43

martinkim0 mentioned this pull request

Remove to_device #85

Open

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels