o
    i                     @   s    d dl Z G dd de jjZdS )    Nc                       sB   e Zd ZdZ					ddedededed	ef
 fd
dZ  ZS )SGDa  Thin inheritance of torch.optim.SGD to bind the required arguments, 'lr'

    Note that
    the arguments of the optimizer invoked by AbsTask.main()
    must have default value except for 'param'.

    I can't understand why only SGD.lr doesn't have the default value.
    皙?        Flrmomentum	dampeningweight_decaynesterovc                    s   t  j||||||d d S )N)r   r   r   r   r	   )super__init__)selfparamsr   r   r   r   r	   	__class__ I/home/ubuntu/.local/lib/python3.10/site-packages/funasr/optimizers/sgd.pyr      s   	
zSGD.__init__)r   r   r   r   F)__name__
__module____qualname____doc__floatboolr   __classcell__r   r   r   r   r      s$    r   )torchoptimr   r   r   r   r   <module>   s    