universal-ie / UIE

Unified Structure Generation for Universal Information Extraction
900 stars 99 forks source link

分布式出现问题 #32

Open Shajiu opened 2 years ago

Shajiu commented 2 years ago

您好!我的环境为如下: 我环境就是按照官网配置的,另外完全按照官网进行运行的,但是还是出现问题,请问具体哪里设置gpu个数和指定GPU啊?

luyaojie commented 2 years ago

这边环境没有显示,执行的是README里面这个命令对吗?

bash run_uie_finetune.bash -v -d 0 \
  -b 16 \
  -k 3 \
  --lr 1e-4 \
  --warmup_ratio 0.06 \
  -i absa/14lap \
  --epoch 50 \
  --spot_noise 0.1 \
  --asoc_noise 0.1 \
  -f spotasoc \
  --epoch 50 \
  --map_config config/offset_map/closest_offset_en.yaml \
  -m hf_models/uie-base-en \
  --random_prompt
luyaojie commented 2 years ago

-d 这个是GPU 的 ID。