mlx.nn.Module.save_weights

mlx.nn.Module.save_weights#

Module.save_weights(file: str)#

将模型的权重保存到文件。保存方法由文件扩展名决定: - .npz 将使用 mx.savez() - .safetensors 将使用 mx.save_safetensors()