# freezeParameters
将 ONNXParameters 中可学习的网络参数转换为不可学习
函数库: TyDeepLearning
# 语法
freezenet = freezeParameters(net, names)
# 说明
net = freezeParameters(net, names) 冻结ONNX模型网络层权重参数。
# 示例
冻结ONNX模型权重参数
加载ONNX模型, 并冻结所有权重参数。
using TyDeepLearning
set_backend(:mindspore)
file = joinpath(pkgdir(TyDeepLearning), "examples/Images/ONNX/net.onnx")
model = importONNXNetwork(file)
冻结模型权重参数。
freezeParameters(model, "all")
freeze complete!
# 输入参数
net-待冻结权重参数网络字符串
输入模型的路径,目前支持导入“ONNX”格式的模型。
names-需冻结网络层名称字符串
需要冻结的网络层名称,默认为“all”。
# 输出参数
freezenet-已冻结网络权重网络网络对象
网络权重参数已冻结,不会再更新。