search

helpmefindaname / transformer-smaller-training-vocab

Temporary remove unused tokens during training to save ram and speed.
https://helpmefindaname.github.io/transformer-smaller-training-vocab/
MIT License
20 stars 2 forks source link

add support for torch 2.0.0 #5

Closed helpmefindaname closed 1 year ago

helpmefindaname commented 1 year ago

Loosen the dependency requirement upwards to support any torch 2.x version