AI4Finance-Foundation / AI4Finance-Foundation/FinRL_Crypto

How to improve this model's train effeciency on GPU

Aberta
#4 0 comentários 0 reações 0 responsáveis Ver no GitHub
help wanted
Linguagem predominante
Python
Estrelas
201
Forks
88
Métricas de merge de PRs
Nenhum PR com merge em 30d

Descrição

I tried to run the part of cpcv of this model on Google Colab Pro+ with A100 GPU but it only uses 1.7GB over 40GB GPU RAM. No matter how I changed the batch size, worker_num of the GPU, thread_num of the CPU. It seems to me that there is a bottleneck that restrains the GPU performance and train speed. Can anyone help with this problem?

Guia de contribuição

Nenhum guia de contribuição indexado para este repositório

Avaliação

Esta issue ainda não foi avaliada.

Receba novas issues na sua caixa de entrada

Um resumo curto de issues do GitHub para quem está começando.