Closed mushroom1116 closed 6 years ago
I received some errors when running the code. OOM when allocating tensor with shape[512,1850,32,32] So I would like to ask how much memory is needed to run this code? Thank you.
OOM when allocating tensor with shape[512,1850,32,32]
Out of the box the project requires around 10.5 GB of GPU memory to run. If that doesn't fit, you have two options:
json
I received some errors when running the code.
OOM when allocating tensor with shape[512,1850,32,32]
So I would like to ask how much memory is needed to run this code? Thank you.