search

cellarium-ai / cellarium-ml

Distributed single-cell data analysis.
BSD 3-Clause "New" or "Revised" License
11 stars 2 forks source link

Set `torch.Generator` device in `reset_parameters` #191

Closed ordabayevy closed 3 months ago

ordabayevy commented 3 months ago

Fixes #190