search

fishaudio / fish-diffusion

An easy to understand TTS / SVS / SVC framework
https://diff.fish.audio
MIT License
635 stars 81 forks source link

fix batch size warn on hifisinger v2 #102

Closed Yurzi closed 1 year ago

Yurzi commented 1 year ago

just fix a warning.

/opt/miniconda/envs/fish/lib/python3.10/site-packages/pytorch_lightning/utilities/data.py:76: UserWarning: Trying to infer the `batch_size` from an ambiguous collection. The batch size we found is 2. To avoid any miscalculations, use `self.log(..., batch_size=batch_size)`.
warning_cache.warn( 
<stacktraces>