Jake Lee

icon indicating an email [email protected]

icon company Amazon AI icon location Taiwan USA icon location Contribute to Apache MXNet, Co-author of DJL (Deep Java Library), Focusing on distributed training now

Results 12 comments of Jake Lee

OOME for https://github.com/kingyuluk/RL-FlappyBird/ on Windows & Ubuntu

@nimishjain15882 could you set MXNET_GPU_MEM_POOL_RESERVE to 10 or 20? MXNET_GPU_MEM_POOL_RESERVE is used to tell MXNet how much memory we want to reserve for cuda kernel launch and cudnn handle space,...

It seems to be some problems when load faster Rcnn by pytorch model

@yang162132 Thanks for reporting the issue. It is expected as the NMS operator is implemented in [TorchVision](https://github.com/pytorch/vision/blob/1ac6e8b91b980b052324f77828a5ef4a6715dd66/torchvision/csrc/ops/nms.cpp#L10) and not part of libtorch.