fsner
fsner copied to clipboard
What is the minimum requirement for a GPU?
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