kongds / HBGL

Exploiting Global and Local Hierarchies for Hierarchical Text Classification
23 stars 3 forks source link

version of pytorch #2

Open RomanPlaud opened 1 year ago

RomanPlaud commented 1 year ago

hi, which version of pytorch are you using ? as you specified transformers=2.10.0 i used pytorch 1.1 but i'm not sure it is the right version

kongds commented 1 year ago

Hi, we use PyTorch 1.7.1 version.