model.state_dict()是做什么的?
173******41 发布于2020-03 浏览:9064 回复:4
0
收藏

教程里说用来保存模型:fluid.save_dygraph(model.state_dict(), 'LR_model'),但是不知道model.state_dict()是什么意思,作用是什么?

收藏
点赞
0
个赞
共4条回复 最后由阿扁门前一棵松回复于2020-03
#5阿扁门前一棵松回复于2020-03

模型的参数字典 存放模型时使用

0
#4189******30回复于2020-03

model.state_dict()这个是模型的参数字典,fluid.save_dygraph()就是save的这个东西。读取就是读取的这个东西。

1
#3西南交大开源回复于2020-03

保存模型参数

0
#2BadDay_回复于2020-03

保存着模型的参数

0
TOP
切换版块