model feature extraction not possible on cuda

Question

model feature extraction not possible on cuda

florianmahner opened this issue 3 years ago · 1 comments

if torch.device is cuda, then thingsvision.model_class.Model.extract_features() throws a type error, since features are torch.Tensor() and features = np.vstack(features) #L223 is not possible. suggested fix on #L221 features.append(act.cpu().numpy())

Answer 1 · 2022-04-13T12:02:58.000Z

Feel free to open a PR. :)