Repository

AI4Finance-Foundation/FinRL

FinRL: Financial Reinforcement Learning. 🔥
7432 1779 177 598
  • 000
I have cuda installed. I have given "device": "cuda" in params but I am still getting using cpu device. A2C_PARAMS = {"n_steps": 5, "ent_coef": 0.01, "learning_rate": 0.0005, "device": "cuda"} output: {'n_steps': 5, 'ent_coef': 0.01, 'learning_rate': 0.0005, 'device': 'cuda'} Using cpu device Loggin...