Ray

Results 3 comments of Ray

Same. 2080TI (11GB) with batch_size = 1 still not work. Here's the traceback: ``` Traceback (most recent call last): File "train.py", line 195, in train() File "train.py", line 140, in...

You can try NVIDIA apex with `opt_level = 'O2`, I got 8100M GPU memory usage with batch size 16, you can try to use smaller batch size to fit in...

> Thanks for reporting this. Would you be interested in contributing a PR to fix this? Sure! I'll submit a PR later.