Published March 20, 2021 | Version 1.1
Software Open

i-machine-think/diagNNose: Improve GPU and Transformer support

Description

  • Comply library with transformers v4.0
  • Improve support for Transformer LMs
  • Improve GPU support
  • Make __getitem__ of ActivationReader more efficient
  • Various minor bug fixes

Files

i-machine-think/diagNNose-1.1.zip

Files (115.9 kB)

Name Size Download all
md5:beccf54a929a35fd94428378afa65545
115.9 kB Preview Download

Additional details

Related works