Improved PyTorch integration
The step
and predict/__call__
functions, GBRL functions in GBRL now work directly with PyTorch tensors, eliminating the need to clone data to the CPU and convert it to NumPy.
The step
and predict/__call__
functions, GBRL functions in GBRL now work directly with PyTorch tensors, eliminating the need to clone data to the CPU and convert it to NumPy.