U
    yh                     @   s2   d dl mZ d dlZdgZG dd dejjZdS )    )
deprecatedN
GradScalerc                	       s>   e Zd ZdZededdeeeeed	d
 fddZ	  Z
S )r   z
    See :class:`torch.amp.GradScaler`.
    ``torch.cpu.amp.GradScaler(args...)`` is deprecated. Please use ``torch.amp.GradScaler("cpu", args...)`` instead.
    zm`torch.cpu.amp.GradScaler(args...)` is deprecated. Please use `torch.amp.GradScaler('cpu', args...)` instead.)category      @       @      ?  TN)
init_scalegrowth_factorbackoff_factorgrowth_intervalenabledreturnc                    s   t  jd|||||d d S )Ncpu)r	   r
   r   r   r   )super__init__)selfr	   r
   r   r   r   	__class__ K/var/www/html/venv/lib/python3.8/site-packages/torch/cpu/amp/grad_scaler.pyr      s    zGradScaler.__init__)r   r   r   r   T)__name__
__module____qualname____doc__r   FutureWarningfloatintboolr   __classcell__r   r   r   r   r      s$        )typing_extensionsr   Ztorch__all__ampr   r   r   r   r   <module>   s   