2024年5月6日
Huggingface Transformers实现张量并行的小坑 set/get_output_embeddings
摘要： transformers 库里实现的很多模型会有这么两个函数 get_output_embeddings和 get_output_embeddings。以 SwitchTransformer 为例 class SwitchTransformersForConditionalGeneration(Sw 阅读全文
posted @ 2024-05-06 11:17 marsggbo 阅读(111) 评论(0) 推荐(0) 编辑