Files
CavinHuang-comfyui-nodes-docs/docs/ADE_AdjustWeightAllMult.md
T
2024-05-26 20:35:16 +08:00

2.7 KiB

Documentation

WeightAdjustAllMultNode类的'get_weight_adjust'方法旨在对模型中的所有权重应用统一的乘数,从而在不改变底层架构的情况下微调模型参数。它是调整模型权重整体规模的关键工具,特别适用于需要根据不同输入范围或灵敏度水平校准模型响应的场景。

Input types

Required

  • all_MULT
    • 参数'all_MULT'对于确定将应用于模型中所有权重的统一乘数至关重要。它允许对模型的权重进行简单的缩放,这可以显著影响模型的性能和行为。此参数对于调整模型对输入数据的敏感度非常关键。
    • Comfy dtype: FLOAT
    • Python dtype: float

Optional

  • print_adjustment
    • 参数'print_adjustment'是一个可选的标志,设置为True时启用记录权重调整的详细信息。这对于调试或希望在调整过程中跟踪对模型权重所做的调整的用户来说非常有用。
    • Comfy dtype: BOOLEAN
    • Python dtype: bool
  • prev_weight_adjust
    • 参数'prev_weight_adjust'允许用户提供一个先前的权重调整组,然后方法将在此基础上构建。这对于增量调整非常有用,其中方法可以从先前建立的状态继续,而不是从头开始。
    • Comfy dtype: WEIGHT_ADJUST
    • Python dtype: AdjustGroup

Output types

  • WEIGHT_ADJUST
    • 'get_weight_adjust'方法的输出是一个'WEIGHT_ADJUST'对象,它封装了对模型权重所做的调整。这个对象非常重要,因为它代表了权重调整过程后模型参数的更新状态,准备用于进一步的模型训练或推理。
    • Comfy dtype: WEIGHT_ADJUST
    • Python dtype: AdjustGroup

Usage tips

  • Infra type: CPU

Source code

class WeightAdjustAllMultNode:

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'all_MULT': ('FLOAT', {'default': 1.0, 'min': 0.0, 'max': 2.0, 'step': 1e-06}), 'print_adjustment': ('BOOLEAN', {'default': False})}, 'optional': {'prev_weight_adjust': ('WEIGHT_ADJUST',)}}
    RETURN_TYPES = ('WEIGHT_ADJUST',)
    CATEGORY = 'Animate Diff 🎭🅐🅓/ad settings/weight adjust'
    FUNCTION = 'get_weight_adjust'

    def get_weight_adjust(self, all_MULT: float, print_adjustment: bool, prev_weight_adjust: AdjustGroup=None):
        if prev_weight_adjust is None:
            prev_weight_adjust = AdjustGroup()
        prev_weight_adjust = prev_weight_adjust.clone()
        adjust = AdjustWeight(all_MULT=all_MULT, print_adjustment=print_adjustment)
        prev_weight_adjust.add(adjust)
        return (prev_weight_adjust,)