mlx.nn.Module.update#
- 模块。更新(参数: 字典) → 模块#
用提供的字典和列表中的参数替换此模块的参数。
通常由优化器使用,以将模型更改为更新(优化)后的参数。也由
mlx.nn.value_and_grad()使用,以设置模型中的追踪器,以便计算梯度。传入的参数字典不需要是一个完整的字典,类似于
parameters()。只有提供的位置会被更新。- Parameters:
参数 (字典) – 模块参数的完整或部分字典。
- Returns:
更新参数后的模块实例。
用提供的字典和列表中的参数替换此模块的参数。
通常由优化器使用,以将模型更改为更新(优化)后的参数。也由mlx.nn.value_and_grad()使用,以设置模型中的追踪器,以便计算梯度。
传入的参数字典不需要是一个完整的字典,类似于parameters()。只有提供的位置会被更新。
参数 (字典) – 模块参数的完整或部分字典。
更新参数后的模块实例。