search

AllenInstitute / cell_type_mapper

Repository for storing prototype functionality implementations for the BKP
Other
11 stars 2 forks source link

Adding GPU support #2

Closed cacerca closed 1 year ago

danielsf commented 1 year ago

@cacerca

It looks like the only function that gets pushed to the GPU is _subtract_mean_and_normalize (the call to torch.matmul() explicitly invokes cpu()). How much of a speed-up does this get us?