search

facebookresearch / FAMBench

Benchmarks to capture important workloads.
Apache License 2.0
28 stars 23 forks source link

port xlmr to use gpu #31

Closed erichan1 closed 2 years ago

erichan1 commented 2 years ago

Changed:

  1. Added torch.cuda.synchronize for timing
  2. Set data and xlmr model to use gpu (cuda)
  3. Set alias of logger to famlogger in preparation for when fb5logger name changes to famlogger