AI4Finance-Foundation / AI4Finance-Foundation/FinRL_Crypto

How to improve this model's train effeciency on GPU

Aperta
#4 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub
help wanted
Lingua principale
Python
Stelle
201
Fork
88
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

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?

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.