文章预览 pytorch冻结模型某层参数 2023年08月22日 539 字 ### pytorch如何冻结某层参数 bert模型12层,参数量达一亿,bert模型做微调有的时候就需要只训练部分参数,那么就需要把其他的参数冻结掉就可以 ``` import torch.nn as nn from transformers import BertModel import torch model = BertModel() … pytorch 深度学习 自然语言 bert bertmodel 优化器 阅读全文 浏览 评论