search

LeeSureman / Batch_Parallel_LatticeLSTM

Chinese NER using Lattice LSTM. Reproduction for ACL 2018 paper.
130 stars 16 forks source link

冒昧請問一下這樣是有在訓練嗎?如果有，為何GPU資源沒有使用到呢? #23

Open arilrve opened 3 years ago

arilrve commented 3 years ago

image image image 環境: GPU : RTX 2060 CUDA : image image

iamlockelightning commented 3 years ago

您可以先看看win上的torch能不能用cuda https://stackoverflow.com/questions/48152674/how-to-check-if-pytorch-is-using-the-gpu

再看看参数传的对不对（main里面的arg print打印靠前。也就是说print之后可能也有arg变化）