mlx.nn.Module.update

mlx.nn.Module.update#

模块。更新(参数 字典) 模块#

用提供的字典和列表中的参数替换此模块的参数。

通常由优化器使用,以将模型更改为更新(优化)后的参数。也由mlx.nn.value_and_grad()使用,以设置模型中的追踪器,以便计算梯度。

传入的参数字典不需要是一个完整的字典,类似于parameters()。只有提供的位置会被更新。

Parameters

参数 (字典) – 模块参数的完整或部分字典。

Returns

更新参数后的模块实例。