- Este tópico contém 1 resposta, 2 vozes e foi atualizado pela última vez 12 meses atrás por
.
Visualizando 2 posts - 1 até 2 (de 2 do total)
- Posts
Olá, relativamente a validação cruzada(pelo menos na classificação binária, que é só onde estou ainda) reparei que é usado o CPU. É possível fazer o mesmo com GPU? Como é que os pesquisadores (por exemplo) fazem este tipo de validação cruzada em datasets com milhões de dados? Há alguma forma de optimizar esta validação?
Obrigado!
Olá Luis,
A biblioteca skorch que usamos para fazer validação cruzada no curso suporta o uso de GPUs, como é documentado aqui. Mesmo assim, é possível implementar a validação cruzada de forma mais manual, criando os folds explicitamente, e aproveitar o suporte a GPUs que o próprio PyTorch já tem.
Visualizando 2 posts - 1 até 2 (de 2 do total)
- Você deve fazer login para responder a este tópico.