What are the versions of torch and transformers

xmshi-trio / MSL

Other

22 stars 4 forks source link

What are the versions of torch and transformers #5

Closed FlyingCat-fa closed 1 year ago

FlyingCat-fa commented 2 years ago

When I run train.py with train_cnn.json: There is a error "dataset attribute should not be set after DataLoader is initialized" in trans_id_to_bert_embedding: data_loader.dataset = output_dataset

Could tell me the versions of torch and transformers ? Thanks.

xmshi-trio commented 2 years ago

The version of torch is 1.0.0, and the version of transformers is 2.7.0