LiuZhenshun

GPU out of memory

1

Hi, I tried to use 8G 2080-ti to evaluate your 152-resnet model, but the memory is not enough. Could you tell the minimum Memory to run your test code?

Problem of Cuda function for prepocessing the batch images from Videosource

9

Hi, Sir, Recently, I would like to inference with batch size bigger than 1, cause i gonna connect several cameras and inference multi batch in once time. I used videosource...

### Description Hi, I would like to install trax locally. Firstly, I found jax I installed is not suitable for GPU, so I follow the jax github to install Cuda...