fsner icon indicating copy to clipboard operation
fsner copied to clipboard

What is the minimum requirement for a GPU?

Open iloveu520 opened this issue 1 year ago • 0 comments

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

iloveu520 avatar May 24 '24 14:05 iloveu520