
    קg1                     \    d dl mZ d dlZd dlmZ dgZ G d dej        j                  ZdS )    )
deprecatedN)OptState
GradScalerc                   n     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      @       @      ?  T
init_scalegrowth_factorbackoff_factorgrowth_intervalenabledreturnNc                 V    t                                          d|||||           d S )Ncuda)r   r   r   r   r   )super__init__)selfr   r   r   r   r   	__class__s         V/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/cuda/amp/grad_scaler.pyr   zGradScaler.__init__   sB     	!')+ 	 	
 	
 	
 	
 	
    )r   r	   r
   r   T)__name__
__module____qualname____doc__r   FutureWarningfloatintboolr   __classcell__)r   s   @r   r   r      s         
 Z	F   $" ##
 

 
 	

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