LiemTr

Dear all, I used CUDA 9.1 and could not run file train.py. I wonder that this code can run on CUDA 9.1 or not? Thank you