reshape error

Describe the bug

Original data size: 800000 Expected shape: (200, 1000, 3) Width x Height: (1000, 200) Epoch 49/999 -------------------- ------------------- 2/4 0:02:08 • -:--:-- 0.00it/s v_num: 0.000 loss/g/total: 36.134 loss/g/fm: 8.935 loss/g/mel: 23.134 loss/g/kl: 1.493 loss/g/lf0: 0.001 loss/d/total: 2.278 Traceback (most recent call last): File "C:\Program Files\Python310\lib\runpy.py", line 196, in _run_module_as_main return _run_code(code, main_globals, None, File "C:\Program Files\Python310\lib\runpy.py", line 86, in run_code exec(code, run_globals) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\Scripts\svc.exe_main.py", line 7, in File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\click\core.py", line 1161, in call return self.main(*args, **kwargs) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\click\core.py", line 1082, in main rv = self.invoke(ctx) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\click\core.py", line 1697, in invoke return _process_result(sub_ctx.command.invoke(sub_ctx)) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\click\core.py", line 1443, in invoke return ctx.invoke(self.callback, **ctx.params) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\click\core.py", line 788, in invoke return _callback(*args, **kwargs) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\so_vits_svc_fork_main.py", line 128, in train train( File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\so_vits_svc_fork\train.py", line 149, in train trainer.fit(model, datamodule=datamodule) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\trainer\trainer.py", line 539, in fit call._call_and_handle_interrupt( File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\trainer\call.py", line 47, in _call_and_handle_interrupt return trainer_fn(*args, **kwargs) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\trainer\trainer.py", line 575, in _fit_impl self._run(model, ckpt_path=ckpt_path) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\trainer\trainer.py", line 982, in _run results = self._run_stage() File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\trainer\trainer.py", line 1026, in _run_stage self.fit_loop.run() File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\loops\fit_loop.py", line 216, in run self.advance() File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\loops\fit_loop.py", line 455, in advance self.epoch_loop.run(self._data_fetcher) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\loops\training_epoch_loop.py", line 150, in run self.advance(data_fetcher) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\loops\training_epoch_loop.py", line 322, in advance batch_output = self.manual_optimization.run(kwargs) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\loops\optimization\manual.py", line 94, in run self.advance(kwargs) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\loops\optimization\manual.py", line 114, in advance training_step_output = call._call_strategy_hook(trainer, "training_step", *kwargs.values()) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\trainer\call.py", line 323, in _call_strategy_hook output = fn(*args, **kwargs) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\lightning\pytorch\strategies\strategy.py", line 391, in training_step return self.lightning_module.training_step(*args, **kwargs) File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\so_vits_svc_fork\train.py", line 483, in training_step "slice/mel_org": utils.plot_spectrogram_to_numpy( File "C:\Users\anbin\AppData\Roaming\so-vits-svc-fork\venv\lib\site-packages\so_vits_svc_fork\\utils.py", line 407, in plot_spectrogram_to_numpy data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,)) ValueError: cannot reshape array of size 800000 into shape (200,1000,3)

To Reproduce

I got this error and hope someone can help

Additional context

No response

Version

4.1

Platform

Win10

Code of Conduct

• I agree to follow this project's Code of Conduct.

No Duplicate

• I have checked existing issues to avoid duplicates.