sayef / fsner

Few-shot Named Entity Recognition
119 stars 6 forks source link

What is the minimum requirement for a GPU? #13

Open iloveu520 opened 3 months ago

iloveu520 commented 3 months ago

hello,I processed the dataset and ran it, but kept killing processes. It's a 40GB A30 with dual cards. This is my running command: Python trainer. py -- pre trained model/home/jovyan/work/Pytorch NER main/machine bet/-- mode train -- train data train. json -- val data val. json -- train batch size 2-- val batch size 2-- n examples per entity 6-- new example batch ratio 1/3-- max epochs 25-- device gpu -- gpus -1-- strategy ddp