#python #matlab #deep-learning #pytorch #conv-neural-network
Вопрос:
Я беру выходные данные некоторых слоев в своей модели и пытаюсь сохранить их в формате .mat. Но матрица пуста! Не мог бы кто-нибудь помочь, пожалуйста. Спасибо
out_recon, my_feat = model(in_imgs)
my_feat = np.array(my_feat)
print("my_feat", my_feat[0].shape)
scipy.io.savemat('test.mat', {'mydata': my_feat[0]})
Размер моего подвига:
факел my_feat.Размер([1, 512, 1, 24, 24])
Комментарии:
1. не могли бы вы, пожалуйста, попробовать проверить с помощью my_feat[0].cpu().numpy()
2. да, это работает для my_feat[0], спасибо! Но теперь, когда я делаю scipy.io.savemat(‘test1.mat’, {‘mydata’: my_feat}), это дает мне пустую матрицу!
3. для этого, пожалуйста, попробуйте аналогичный my_feat.cpu().numpy() . если это сработает, пожалуйста, скажите мне, что я добавлю это в качестве ответа
4. когда я делаю my_feat.cpu().numpy(), он выдает ошибку атрибута: объект ‘numpy.ndarray’ не имеет атрибута ‘cpu’
5. my_feat = np.array(my_feat) , не могли бы вы прокомментировать эту строку и попробовать один раз