Open kids0cn opened 4 years ago
Our test environment was CUDA10, Python 3.6 and PyTorch 0.4 and 1.0. However, one of my colleagues tested the code with CUDA 9 and it worked well. Since the error in your log is an OOM error, I suggest you check those hyper-parameters first, e. g., the batch size. Also, this code has not yet been tested on Python 3.7.
Hi,I have the same question.I try reproduce the code on CUDA 9 python 3.6 pytorch 0.4,but it still out of memory.
Hi there, I try to reproduce your code on NVIDIA V100,CUDA 9,but it can not work。