pytorch-transformers-classification icon indicating copy to clipboard operation
pytorch-transformers-classification copied to clipboard

Published 20 hours ago • verified publisher icon ThilinaRajapakse

Update prerec packages info: transformer and apex?

Open HaoLi111 opened this issue 1 year ago • 1 comments

  1. replacing every pytorch_transformers by transformers fixes the problem of giving negative dimensions to create tensor.
  2. fp 16 requires nv apex, but can you write something to note that it is not the pip apex, and point to a way to install from nv or just let the default fp16=False to circumvent the issue for beginners? Thanks

HaoLi111 avatar Apr 18 '23 02:04 HaoLi111

maybe 2 is not a good idea. Maybe include the code to install apex and use fp16 to save cost?

HaoLi111 avatar Apr 18 '23 02:04 HaoLi111