o
    i@                  
   @   s   d dl Z d dlmZ d dlmZmZ eddZeeZdZdZ	dZ
d	Zeee	e
d
ddZeeee Zede de de	 de
 d	 ede de de ee  d dS )    N)SequentialSampler)DynamicBatchSamplerload_datasetEmilia_ZH_ENpinyin   i   @   i i  F)max_samplesrandom_seeddrop_residualzOne epoch has z updates if gpus=z, with batch_size_per_gpu=z  (frames) & max_samples_per_gpu=.zIf gpus=z, for max_updates=z should set epoch=)mathtorch.utils.datar   f5_tts.model.datasetr   r   train_datasetsamplergpusbatch_size_per_gpumax_samples_per_gpumax_updatesbatch_samplerintlenupdates_per_epochprintceil r   r   Z/home/ubuntu/.local/lib/python3.10/site-packages/f5_tts/scripts/count_max_epoch_precise.py<module>   s2    
*