任务¶
此处给出了MindFormers套件中支持的任务名称和模型名称,用于高阶开发时的索引名
模型 |
任务(task name) |
模型(model name) |
|---|---|---|
masked_language_modeling |
bert_base_uncased |
|
translation |
t5_small |
|
text_generation |
gpt2_small |
|
text_generation |
pangualpha_2_6_b |
|
text_generation |
glm_6b |
|
text_generation |
llama_7b |
|
text_generation |
bloom_560m |
|
masked_image_modeling |
mae_vit_base_p16 |
|
vit_base_p16 |
||
swin_base_p4w7 |
||
contrastive_language_image_pretrain |
clip_vit_b_32 |