Closed wkcn closed 4 years ago
This PR will fix the bug which crash when training model on multiple GPUs. #62
The reason is that NDArrays are released before the function runs.
However, this PR will allocate the memory before the function runs rather than during the function runs. I will try to find a better solution.
This PR will fix the bug which crash when training model on multiple GPUs. #62
The reason is that NDArrays are released before the function runs.
However, this PR will allocate the memory before the function runs rather than during the function runs. I will try to find a better solution.