# Attention! check if EMBEDDING LAYER is trained ! e.g. bert.embeddings.word_embeddings.weight
for name, param in self.model.named_parameters():
if 'embeddings' in name and not param.requires_grad:
param.requires_grad = True
## 检查emb层是否参与训练
[linux] bert预训练,检查emb是否参与训练
最新推荐文章于 2024-07-07 01:02:21 发布
本文详细介绍了在Linux操作系统中进行BERT预训练的过程,探讨如何验证 Embedding 层是否参与了训练。内容包括使用Python实现训练脚本,深度学习库的运用,以及相关命令行工具的使用技巧。
摘要由CSDN通过智能技术生成