search

moskomule / anatome

Ἀνατομή is a PyTorch library to analyze representation of neural networks
MIT License
61 stars 6 forks source link

when is cca faster on cpu than gpu? #5

Closed brando90 closed 3 years ago

brando90 commented 3 years ago

I saw:

https://github.com/moskomule/anatome/blob/57f34fa796ffcff0ba37b5fa5142018e9f6fde61/anatome/similarity.py#L223

can this be clarified? when is this the case?

moskomule commented 3 years ago

Previously, SVD on GPU was extremely slower than on CPU. It depends on the cases.