Quando tento rodar esta parte do código, para a extração de características, tenho o seguinte erro:
print('Load train')
train_X, train_Y = extrai_caracteristicas(net, train_loader)
print('\nLoad test')
test_X, test_Y = extrai_caracteristicas(net, test_loader)RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
Não consegui achar o motivo deste erro. Alguém consegue me ajudar?
