-
Install PyTorch https://pytorch.org/get-started/locally/
Make sure that the CUDA version is same as your computer.
Use
nvcc --version
to check your CUDA version. -
Install apex according to this link: https://github.com/nvidia/apex
-
pip install
fast_bert
-
Change the route of the folder.
*DATA_PATH* = Path('/home/mist/Test/Bert/data/') *LABEL_PATH* = Path('/home/mist/Test/Bert/labels/') *AUG_DATA_PATH* = Path('/home/mist/Test/Bert/data/data_augmentation/') *MODEL_PATH*=Path('/home/mist/Test/Bert/models/') *LOG_PATH*=Path('/home/mist/Test/Bert/logs/')
-
Run the code