提交 9a5809fa 编写于 作者: B bbking 提交者: pkpk

add cuda_check in PaddleNLP emotion_detection (#2731)

* add cuda_check

* add cuda_check in emotion_detection
上级 b7581294
......@@ -18,6 +18,7 @@ import paddle.fluid as fluid
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)
......@@ -19,6 +19,7 @@ import numpy as np
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)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册