diff --git a/demo/ernie-seq-labeling/run_fintune_with_hub.sh b/demo/ernie-seq-labeling/run_fintune_with_hub.sh index bf4a10f2013f6556afb088c55206882911969cdb..92bafcb75693600e2ccf4ea3deba458362e911f5 100644 --- a/demo/ernie-seq-labeling/run_fintune_with_hub.sh +++ b/demo/ernie-seq-labeling/run_fintune_with_hub.sh @@ -1,4 +1,4 @@ -export CUDA_VISIBLE_DEVICES=5 +export CUDA_VISIBLE_DEVICES=0 CKPT_DIR="./ckpt" diff --git a/requirements.txt b/requirements.txt index 8dc37c121be5797424f469802e0ef081536e8772..1c20445c5e609acafdaaa10e3c4c6e0c43acc3c2 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ paddlepaddle >= 1.3.0 +visualdl >= 1.3.0 pre-commit protobuf >= 3.1.0 yapf == 0.26.0