Input Data Mismatch

[Lijingze666]

When I use my own data to train the model, the following error is prompted. What changes do I need to make to the input data to run the program normally?
Traceback (most recent call last):
File "/home/yangjingwen/spiking-fullsubnet-main/recipes/intel_ndns/spiking_fullsubnet_freeze_phase/run.py", line 185, in
run(config, args.resume)
File "/home/yangjingwen/spiking-fullsubnet-main/recipes/intel_ndns/spiking_fullsubnet_freeze_phase/run.py", line 127, in run
trainer.train(train_dataloader, validate_dataloaders)
File "/home/yangjingwen/spiking-fullsubnet-main/recipes/intel_ndns/spiking_fullsubnet_freeze_phase/audiozen/trainer/base_trainer_gan_accelerate_ddp_validate.py", line 280, in train
loss_dict = self.training_step(batch, batch_idx)
File "/home/yangjingwen/spiking-fullsubnet-main/recipes/intel_ndns/spiking_fullsubnet_freeze_phase/trainer.py", line 63, in training_step
enhanced_y, enhanced_mag, *_ = self.model_g(noisy_y)
File "/home/yangjingwen/anaconda3/envs/spiking-fullsubnet/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/yangjingwen/anaconda3/envs/spiking-fullsubnet/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/home/yangjingwen/spiking-fullsubnet-main/recipes/intel_ndns/spiking_fullsubnet_freeze_phase/model_low_freq.py", line 588, in forward
df_coefs_list, sb_all_layer_outputs = self.sb_model(noisy_mag, fb_output)
File "/home/yangjingwen/anaconda3/envs/spiking-fullsubnet/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/yangjingwen/anaconda3/envs/spiking-fullsubnet/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/home/yangjingwen/spiking-fullsubnet-main/recipes/intel_ndns/spiking_fullsubnet_freeze_phase/model_low_freq.py", line 477, in forward
sb_model_input = self.norm(sb_model_input)
File "/home/yangjingwen/spiking-fullsubnet-main/recipes/intel_ndns/spiking_fullsubnet_freeze_phase/model_low_freq.py", line 182, in cumulative_laplace_norm
batch_size, num_channels, num_freqs, num_frames = input.size()
ValueError: too many values to unpack (expected 4)