Skip to content

BERT/ALBERT based model for RACE dataset, support multi-worker, multi-GPU, FP16 and bind CPU.

Notifications You must be signed in to change notification settings

xcmyz/bert-race

Repository files navigation

BERT for RACE dataset

Run

  1. multi-worker, multi-GPU (blind cpu):
    • bash run_multiworker.sh 0 <addr> 0 1 <model name> <dataset name> 320 <batch size on single GPU>
  2. evaluation:
    • bash eval.sh

About

BERT/ALBERT based model for RACE dataset, support multi-worker, multi-GPU, FP16 and bind CPU.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published