2024-09-28 01:32:30 发布
网友
我尝试使用DeepPavlov库中的ELMO嵌入(ElmoEmbedder)。它的工作非常慢,每100个句子64秒。在
我试图增加mini_batch_的大小,但它并没有加快算法的速度。在
有没有可能加快艾尔莫姆贝德的速度?在
mini_batch_size只有在GPU上运行模型时,才会对性能产生显著影响。在
mini_batch_size
如果没有GPU,请尝试使用Google Colab。它为您提供对GPU和TPU的免费访问。在
!pip install deeppavlov from deeppavlov.deep import find_config from deeppavlov.core.commands.infer import build_model config_path = find_config('elmo_ru-wiki') model = build_model(config_path, load_trained = True, download = True) a = model(data)
不要忘记在Edit->Notebook settings->Hardware accelerator中启用GPU
Edit->Notebook settings->Hardware accelerator
mini_batch_size
只有在GPU上运行模型时,才会对性能产生显著影响。在如果没有GPU,请尝试使用Google Colab。它为您提供对GPU和TPU的免费访问。在
不要忘记在
Edit->Notebook settings->Hardware accelerator
中启用GPU相关问题 更多 >
编程相关推荐