search

jefferyYu / UMT

Preprocessed Datasets for our Multimodal NER paper
112 stars 13 forks source link

请教 #15

Open pipijiev12 opened 2 years ago

pipijiev12 commented 2 years ago

我想问一下 你这个程序运行gpu大概是多大 因为我gpu才只有6g跑不起来

jefferyYu commented 2 years ago

印象中应该是11GB左右。如果跑不起来，可以把batch_size设置小一点。