about train

[yiyihao2000]

Your work is quite impressive. May I ask what version of pytorch you used during the training process to obtain the weight of the linetransformer? I am now trying to use the method and dataset you provided for training, but after loading the new weight, an error is reported, which is missing "RuntimeError: Error (s) in loading state_dict for LineTransformer:
Could you please provide me with some guidance? thanks

[yosungho]

the code was tested with PyTorch 1.8.