reshape error

[KanielOutisSs]

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.