cryodrgn icon indicating copy to clipboard operation
cryodrgn copied to clipboard
verified publisher icon zhonge

Clean up --device specification

Open zhonge opened this issue 2 years ago • 0 comments

Some of the cryodrgn scripts accept --device [int] to specify which GPU to run on, but we should instead support the official torch.device spec: https://pytorch.org/docs/stable/tensor_attributes.html#torch.device

zhonge avatar Feb 10 '23 20:02 zhonge