Fix bug: load model on CPU

jiesutd / LatticeLSTM

Chinese NER using Lattice LSTM. Code for ACL 2018 paper.

1.8k stars 453 forks source link

Closed daniyuu closed 6 years ago

daniyuu commented 6 years ago

Fix bug: map_location is a paramater of torch.load, not the load_state_dict()

I have already tested it at local.

jiesutd commented 6 years ago

Thanks @daniyuu for your commits and test. Already merged.