- This topic has 1 reply, 2 voices, and was last updated 2 years, 10 months ago by
.
Viewing 2 posts - 1 through 2 (of 2 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.
Viewing 2 posts - 1 through 2 (of 2 total)
- You must be logged in to reply to this topic.