add cuda_check in PaddleNLP emotion_detection (#2731)

* add cuda_check

* add cuda_check in emotion_detection

Author: chenbjin

PaddleNLP/emotion_detection/run_classifier.py

@@ -18,6 +18,7 @@
 import numpy as np
 
 from models.classification import nets
+from models.model_check import check_cuda
 import reader
 import config
 import utils
@@ -374,4 +375,5 @@ def get_cards():
 
 if __name__ == "__main__":
     utils.print_arguments(args)
+    check_cuda(args.use_cuda)
     main(args)

PaddleNLP/emotion_detection/run_ernie_classifier.py

@@ -19,6 +19,7 @@
 
 from preprocess.ernie import task_reader
 from models.representation import ernie
+from models.model_check import check_cuda
 import utils
 
 # yapf: disable
@@ -384,4 +385,5 @@ def main(args):
 
 if __name__ == "__main__":
     utils.print_arguments(args)
+    check_cuda(args.use_cuda)
     main(args)