
    sg1                     h    d dl mZ d dlZd dlmZ dgZ G d dej                  j                        Zy)    )
deprecatedN)OptState
GradScalerc                   b     e Zd ZdZ ede      	 	 	 	 	 ddededededed	d
f fd       Z	 xZ
S )r   z
    See :class:`torch.amp.GradScaler`.
    ``torch.cuda.amp.GradScaler(args...)`` is deprecated. Please use ``torch.amp.GradScaler("cuda", args...)`` instead.
    zo`torch.cuda.amp.GradScaler(args...)` is deprecated. Please use `torch.amp.GradScaler('cuda', args...)` instead.)category
init_scalegrowth_factorbackoff_factorgrowth_intervalenabledreturnNc                 0    t         |   d|||||       y )Ncuda)r   r	   r
   r   r   )super__init__)selfr   r	   r
   r   r   	__class__s         M/var/www/html/venv/lib/python3.12/site-packages/torch/cuda/amp/grad_scaler.pyr   zGradScaler.__init__   s)     	!')+ 	 	
    )g      @g       @g      ?i  T)__name__
__module____qualname____doc__r   FutureWarningfloatintboolr   __classcell__)r   s   @r   r   r      su    
 	F $" ##

 
 	

 
 
 



r   )typing_extensionsr   torchtorch.amp.grad_scalerr   __all__ampr    r   r   <module>r%      s/    (  + .
%% 
r   