o
    iQ                     @   s   d Z ed ed dZdZdZdZdZdZd	Zee e Z	e	e e d
 Z
ee
 Zee Zededdee dde d ededd ededd ede	dde
dd dS )zADAPTIVE BATCH SIZEzNAdaptive batch size: using grouping batch sampler, frames_per_gpu fixed fed inzC  -> least padding, gather wavs with accumulated frames in a batch
i2t    i]  iO    i      i  zepochs should be set to: z.0fz (z.1fz x gd_acum )z#progress_bar should show approx. 0/z updatesztotal z hourszmini-batch of z	 frames, z.2fz hours per mini-batchN)__doc__printtotal_hoursmel_hop_lengthmel_sampling_ratewanted_max_updatesgpusframes_per_gpu
grad_accummini_batch_framesmini_batch_hoursupdates_per_epochepochs r   r   R/home/ubuntu/.local/lib/python3.10/site-packages/f5_tts/scripts/count_max_epoch.py<module>   s$    $