Pytorch模型训练中的变量
突发奇想,想看pytorch中某个变量更新状态,
在model初始化时:
emb = []
# 初始化模型参数
for i,model in enumerate(models):
for j,param in enumerate(model.parameters()):
if i==0 and j==1:
emb = param
print("************",emb.shape)
在模型反向传播后
原创
2022-01-05 20:53:38 ·
902 阅读 ·
0 评论