Error during processing

#1
by Kveris - opened

I decided to try your model, but an error occurred during the process.

AssertionError: "stereo needs to be set to True if passing in audio signal that is stereo (channel dimension of 2). also need to be False if mono (channel dimension of 1)"

When the parameter was changed from true to false, this happened.
I tried processing files with one channel and with two.

CUDA is available, use --force_cpu to disable it.
Using device: cuda:0
Windows OS detected, using math or mem efficient attention if input tensor is on cuda
Start from checkpoint: C:\Users\Kveris\Downloads\gilliaan_melroformer_stereoizer_alpha1.ckpt
Traceback (most recent call last):
File "C:\Program Files\Music Source Separation Training 1.0.12\inference.py", line 184, in
proc_folder(None)
File "C:\Program Files\Music Source Separation Training 1.0.12\inference.py", line 169, in proc_folder
model.load_state_dict(state_dict, strict=False)
File "C:\Users\Kveris\AppData\Local\Programs\Python\Python311\Lib\site-packages\torch\nn\modules\module.py", line 2639, in load_state_dict
raise RuntimeError(
RuntimeError: Error(s) in loading state_dict for MelBandRoformer:
size mismatch for band_split.to_features.0.0.gamma: copying a param with shape torch.Size([14]) from checkpoint, the shape in current model is torch.Size([28]).
size mismatch for band_split.to_features.0.1.weight: copying a param with shape torch.Size([256, 14]) from checkpoint, the shape in current model is torch.Size([256, 28]).
size mismatch for band_split.to_features.1.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.1.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.2.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.2.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.3.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.3.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.4.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.4.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.5.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.5.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.6.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.6.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.7.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.7.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.8.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.8.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.9.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.9.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.10.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.10.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.11.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.11.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.12.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.12.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.13.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.13.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.14.0.gamma: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([24]).
size mismatch for band_split.to_features.14.1.weight: copying a param with shape torch.Size([256, 12]) from checkpoint, the shape in current model is torch.Size([256, 24]).
size mismatch for band_split.to_features.15.0.gamma: copying a param with shape torch.Size([14]) from checkpoint, the shape in current model is torch.Size([28]).
size mismatch for band_split.to_features.15.1.weight: copying a param with shape torch.Size([256, 14]) from checkpoint, the shape in current model is torch.Size([256, 28]).
size mismatch for band_split.to_features.16.0.gamma: copying a param with shape torch.Size([14]) from checkpoint, the shape in current model is torch.Size([28]).
size mismatch for band_split.to_features.16.1.weight: copying a param with shape torch.Size([256, 14]) from checkpoint, the shape in current model is torch.Size([256, 28]).
size mismatch for band_split.to_features.17.0.gamma: copying a param with shape torch.Size([14]) from checkpoint, the shape in current model is torch.Size([28]).
size mismatch for band_split.to_features.17.1.weight: copying a param with shape torch.Size([256, 14]) from checkpoint, the shape in current model is torch.Size([256, 28]).
size mismatch for band_split.to_features.18.0.gamma: copying a param with shape torch.Size([18]) from checkpoint, the shape in current model is torch.Size([36]).
size mismatch for band_split.to_features.18.1.weight: copying a param with shape torch.Size([256, 18]) from checkpoint, the shape in current model is torch.Size([256, 36]).
size mismatch for band_split.to_features.19.0.gamma: copying a param with shape torch.Size([18]) from checkpoint, the shape in current model is torch.Size([36]).
size mismatch for band_split.to_features.19.1.weight: copying a param with shape torch.Size([256, 18]) from checkpoint, the shape in current model is torch.Size([256, 36]).
size mismatch for band_split.to_features.20.0.gamma: copying a param with shape torch.Size([18]) from checkpoint, the shape in current model is torch.Size([36]).
size mismatch for band_split.to_features.20.1.weight: copying a param with shape torch.Size([256, 18]) from checkpoint, the shape in current model is torch.Size([256, 36]).
size mismatch for band_split.to_features.21.0.gamma: copying a param with shape torch.Size([20]) from checkpoint, the shape in current model is torch.Size([40]).
size mismatch for band_split.to_features.21.1.weight: copying a param with shape torch.Size([256, 20]) from checkpoint, the shape in current model is torch.Size([256, 40]).
size mismatch for band_split.to_features.22.0.gamma: copying a param with shape torch.Size([20]) from checkpoint, the shape in current model is torch.Size([40]).
size mismatch for band_split.to_features.22.1.weight: copying a param with shape torch.Size([256, 20]) from checkpoint, the shape in current model is torch.Size([256, 40]).
size mismatch for band_split.to_features.23.0.gamma: copying a param with shape torch.Size([22]) from checkpoint, the shape in current model is torch.Size([44]).
size mismatch for band_split.to_features.23.1.weight: copying a param with shape torch.Size([256, 22]) from checkpoint, the shape in current model is torch.Size([256, 44]).
size mismatch for band_split.to_features.24.0.gamma: copying a param with shape torch.Size([26]) from checkpoint, the shape in current model is torch.Size([52]).
size mismatch for band_split.to_features.24.1.weight: copying a param with shape torch.Size([256, 26]) from checkpoint, the shape in current model is torch.Size([256, 52]).
size mismatch for band_split.to_features.25.0.gamma: copying a param with shape torch.Size([26]) from checkpoint, the shape in current model is torch.Size([52]).
size mismatch for band_split.to_features.25.1.weight: copying a param with shape torch.Size([256, 26]) from checkpoint, the shape in current model is torch.Size([256, 52]).
size mismatch for band_split.to_features.26.0.gamma: copying a param with shape torch.Size([26]) from checkpoint, the shape in current model is torch.Size([52]).
size mismatch for band_split.to_features.26.1.weight: copying a param with shape torch.Size([256, 26]) from checkpoint, the shape in current model is torch.Size([256, 52]).
size mismatch for band_split.to_features.27.0.gamma: copying a param with shape torch.Size([30]) from checkpoint, the shape in current model is torch.Size([60]).
size mismatch for band_split.to_features.27.1.weight: copying a param with shape torch.Size([256, 30]) from checkpoint, the shape in current model is torch.Size([256, 60]).
size mismatch for band_split.to_features.28.0.gamma: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([64]).
size mismatch for band_split.to_features.28.1.weight: copying a param with shape torch.Size([256, 32]) from checkpoint, the shape in current model is torch.Size([256, 64]).
size mismatch for band_split.to_features.29.0.gamma: copying a param with shape torch.Size([34]) from checkpoint, the shape in current model is torch.Size([68]).
size mismatch for band_split.to_features.29.1.weight: copying a param with shape torch.Size([256, 34]) from checkpoint, the shape in current model is torch.Size([256, 68]).
size mismatch for band_split.to_features.30.0.gamma: copying a param with shape torch.Size([38]) from checkpoint, the shape in current model is torch.Size([76]).
size mismatch for band_split.to_features.30.1.weight: copying a param with shape torch.Size([256, 38]) from checkpoint, the shape in current model is torch.Size([256, 76]).
size mismatch for band_split.to_features.31.0.gamma: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([80]).
size mismatch for band_split.to_features.31.1.weight: copying a param with shape torch.Size([256, 40]) from checkpoint, the shape in current model is torch.Size([256, 80]).
size mismatch for band_split.to_features.32.0.gamma: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([80]).
size mismatch for band_split.to_features.32.1.weight: copying a param with shape torch.Size([256, 40]) from checkpoint, the shape in current model is torch.Size([256, 80]).
size mismatch for band_split.to_features.33.0.gamma: copying a param with shape torch.Size([44]) from checkpoint, the shape in current model is torch.Size([88]).
size mismatch for band_split.to_features.33.1.weight: copying a param with shape torch.Size([256, 44]) from checkpoint, the shape in current model is torch.Size([256, 88]).
size mismatch for band_split.to_features.34.0.gamma: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([96]).
size mismatch for band_split.to_features.34.1.weight: copying a param with shape torch.Size([256, 48]) from checkpoint, the shape in current model is torch.Size([256, 96]).
size mismatch for band_split.to_features.35.0.gamma: copying a param with shape torch.Size([52]) from checkpoint, the shape in current model is torch.Size([104]).
size mismatch for band_split.to_features.35.1.weight: copying a param with shape torch.Size([256, 52]) from checkpoint, the shape in current model is torch.Size([256, 104]).
size mismatch for band_split.to_features.36.0.gamma: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([112]).
size mismatch for band_split.to_features.36.1.weight: copying a param with shape torch.Size([256, 56]) from checkpoint, the shape in current model is torch.Size([256, 112]).
size mismatch for band_split.to_features.37.0.gamma: copying a param with shape torch.Size([58]) from checkpoint, the shape in current model is torch.Size([116]).
size mismatch for band_split.to_features.37.1.weight: copying a param with shape torch.Size([256, 58]) from checkpoint, the shape in current model is torch.Size([256, 116]).
size mismatch for band_split.to_features.38.0.gamma: copying a param with shape torch.Size([62]) from checkpoint, the shape in current model is torch.Size([124]).
size mismatch for band_split.to_features.38.1.weight: copying a param with shape torch.Size([256, 62]) from checkpoint, the shape in current model is torch.Size([256, 124]).
size mismatch for band_split.to_features.39.0.gamma: copying a param with shape torch.Size([66]) from checkpoint, the shape in current model is torch.Size([132]).
size mismatch for band_split.to_features.39.1.weight: copying a param with shape torch.Size([256, 66]) from checkpoint, the shape in current model is torch.Size([256, 132]).
size mismatch for band_split.to_features.40.0.gamma: copying a param with shape torch.Size([72]) from checkpoint, the shape in current model is torch.Size([144]).
size mismatch for band_split.to_features.40.1.weight: copying a param with shape torch.Size([256, 72]) from checkpoint, the shape in current model is torch.Size([256, 144]).
size mismatch for band_split.to_features.41.0.gamma: copying a param with shape torch.Size([78]) from checkpoint, the shape in current model is torch.Size([156]).
size mismatch for band_split.to_features.41.1.weight: copying a param with shape torch.Size([256, 78]) from checkpoint, the shape in current model is torch.Size([256, 156]).
size mismatch for band_split.to_features.42.0.gamma: copying a param with shape torch.Size([82]) from checkpoint, the shape in current model is torch.Size([164]).
size mismatch for band_split.to_features.42.1.weight: copying a param with shape torch.Size([256, 82]) from checkpoint, the shape in current model is torch.Size([256, 164]).
size mismatch for band_split.to_features.43.0.gamma: copying a param with shape torch.Size([88]) from checkpoint, the shape in current model is torch.Size([176]).
size mismatch for band_split.to_features.43.1.weight: copying a param with shape torch.Size([256, 88]) from checkpoint, the shape in current model is torch.Size([256, 176]).
size mismatch for band_split.to_features.44.0.gamma: copying a param with shape torch.Size([94]) from checkpoint, the shape in current model is torch.Size([188]).
size mismatch for band_split.to_features.44.1.weight: copying a param with shape torch.Size([256, 94]) from checkpoint, the shape in current model is torch.Size([256, 188]).
size mismatch for band_split.to_features.45.0.gamma: copying a param with shape torch.Size([100]) from checkpoint, the shape in current model is torch.Size([200]).
size mismatch for band_split.to_features.45.1.weight: copying a param with shape torch.Size([256, 100]) from checkpoint, the shape in current model is torch.Size([256, 200]).
size mismatch for band_split.to_features.46.0.gamma: copying a param with shape torch.Size([108]) from checkpoint, the shape in current model is torch.Size([216]).
size mismatch for band_split.to_features.46.1.weight: copying a param with shape torch.Size([256, 108]) from checkpoint, the shape in current model is torch.Size([256, 216]).
size mismatch for band_split.to_features.47.0.gamma: copying a param with shape torch.Size([114]) from checkpoint, the shape in current model is torch.Size([228]).
size mismatch for band_split.to_features.47.1.weight: copying a param with shape torch.Size([256, 114]) from checkpoint, the shape in current model is torch.Size([256, 228]).
size mismatch for band_split.to_features.48.0.gamma: copying a param with shape torch.Size([122]) from checkpoint, the shape in current model is torch.Size([244]).
size mismatch for band_split.to_features.48.1.weight: copying a param with shape torch.Size([256, 122]) from checkpoint, the shape in current model is torch.Size([256, 244]).
size mismatch for band_split.to_features.49.0.gamma: copying a param with shape torch.Size([132]) from checkpoint, the shape in current model is torch.Size([264]).
size mismatch for band_split.to_features.49.1.weight: copying a param with shape torch.Size([256, 132]) from checkpoint, the shape in current model is torch.Size([256, 264]).
size mismatch for band_split.to_features.50.0.gamma: copying a param with shape torch.Size([142]) from checkpoint, the shape in current model is torch.Size([284]).
size mismatch for band_split.to_features.50.1.weight: copying a param with shape torch.Size([256, 142]) from checkpoint, the shape in current model is torch.Size([256, 284]).
size mismatch for band_split.to_features.51.0.gamma: copying a param with shape torch.Size([152]) from checkpoint, the shape in current model is torch.Size([304]).
size mismatch for band_split.to_features.51.1.weight: copying a param with shape torch.Size([256, 152]) from checkpoint, the shape in current model is torch.Size([256, 304]).
size mismatch for band_split.to_features.52.0.gamma: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([320]).
size mismatch for band_split.to_features.52.1.weight: copying a param with shape torch.Size([256, 160]) from checkpoint, the shape in current model is torch.Size([256, 320]).
size mismatch for band_split.to_features.53.0.gamma: copying a param with shape torch.Size([172]) from checkpoint, the shape in current model is torch.Size([344]).
size mismatch for band_split.to_features.53.1.weight: copying a param with shape torch.Size([256, 172]) from checkpoint, the shape in current model is torch.Size([256, 344]).
size mismatch for band_split.to_features.54.0.gamma: copying a param with shape torch.Size([186]) from checkpoint, the shape in current model is torch.Size([372]).
size mismatch for band_split.to_features.54.1.weight: copying a param with shape torch.Size([256, 186]) from checkpoint, the shape in current model is torch.Size([256, 372]).
size mismatch for band_split.to_features.55.0.gamma: copying a param with shape torch.Size([198]) from checkpoint, the shape in current model is torch.Size([396]).
size mismatch for band_split.to_features.55.1.weight: copying a param with shape torch.Size([256, 198]) from checkpoint, the shape in current model is torch.Size([256, 396]).
size mismatch for band_split.to_features.56.0.gamma: copying a param with shape torch.Size([210]) from checkpoint, the shape in current model is torch.Size([420]).
size mismatch for band_split.to_features.56.1.weight: copying a param with shape torch.Size([256, 210]) from checkpoint, the shape in current model is torch.Size([256, 420]).
size mismatch for band_split.to_features.57.0.gamma: copying a param with shape torch.Size([226]) from checkpoint, the shape in current model is torch.Size([452]).
size mismatch for band_split.to_features.57.1.weight: copying a param with shape torch.Size([256, 226]) from checkpoint, the shape in current model is torch.Size([256, 452]).
size mismatch for band_split.to_features.58.0.gamma: copying a param with shape torch.Size([244]) from checkpoint, the shape in current model is torch.Size([488]).
size mismatch for band_split.to_features.58.1.weight: copying a param with shape torch.Size([256, 244]) from checkpoint, the shape in current model is torch.Size([256, 488]).
size mismatch for band_split.to_features.59.0.gamma: copying a param with shape torch.Size([260]) from checkpoint, the shape in current model is torch.Size([520]).
size mismatch for band_split.to_features.59.1.weight: copying a param with shape torch.Size([256, 260]) from checkpoint, the shape in current model is torch.Size([256, 520]).
size mismatch for mask_estimators.0.to_freqs.0.0.4.weight: copying a param with shape torch.Size([28, 512]) from checkpoint, the shape in current model is torch.Size([56, 512]).
size mismatch for mask_estimators.0.to_freqs.0.0.4.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([56]).
size mismatch for mask_estimators.0.to_freqs.1.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.1.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.2.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.2.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.3.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.3.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.4.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.4.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.5.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.5.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.6.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.6.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.7.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.7.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.8.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.8.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.9.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.9.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.10.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.10.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.11.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.11.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.12.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.12.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.13.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.13.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.14.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.0.to_freqs.14.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.0.to_freqs.15.0.4.weight: copying a param with shape torch.Size([28, 512]) from checkpoint, the shape in current model is torch.Size([56, 512]).
size mismatch for mask_estimators.0.to_freqs.15.0.4.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([56]).
size mismatch for mask_estimators.0.to_freqs.16.0.4.weight: copying a param with shape torch.Size([28, 512]) from checkpoint, the shape in current model is torch.Size([56, 512]).
size mismatch for mask_estimators.0.to_freqs.16.0.4.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([56]).
size mismatch for mask_estimators.0.to_freqs.17.0.4.weight: copying a param with shape torch.Size([28, 512]) from checkpoint, the shape in current model is torch.Size([56, 512]).
size mismatch for mask_estimators.0.to_freqs.17.0.4.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([56]).
size mismatch for mask_estimators.0.to_freqs.18.0.4.weight: copying a param with shape torch.Size([36, 512]) from checkpoint, the shape in current model is torch.Size([72, 512]).
size mismatch for mask_estimators.0.to_freqs.18.0.4.bias: copying a param with shape torch.Size([36]) from checkpoint, the shape in current model is torch.Size([72]).
size mismatch for mask_estimators.0.to_freqs.19.0.4.weight: copying a param with shape torch.Size([36, 512]) from checkpoint, the shape in current model is torch.Size([72, 512]).
size mismatch for mask_estimators.0.to_freqs.19.0.4.bias: copying a param with shape torch.Size([36]) from checkpoint, the shape in current model is torch.Size([72]).
size mismatch for mask_estimators.0.to_freqs.20.0.4.weight: copying a param with shape torch.Size([36, 512]) from checkpoint, the shape in current model is torch.Size([72, 512]).
size mismatch for mask_estimators.0.to_freqs.20.0.4.bias: copying a param with shape torch.Size([36]) from checkpoint, the shape in current model is torch.Size([72]).
size mismatch for mask_estimators.0.to_freqs.21.0.4.weight: copying a param with shape torch.Size([40, 512]) from checkpoint, the shape in current model is torch.Size([80, 512]).
size mismatch for mask_estimators.0.to_freqs.21.0.4.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([80]).
size mismatch for mask_estimators.0.to_freqs.22.0.4.weight: copying a param with shape torch.Size([40, 512]) from checkpoint, the shape in current model is torch.Size([80, 512]).
size mismatch for mask_estimators.0.to_freqs.22.0.4.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([80]).
size mismatch for mask_estimators.0.to_freqs.23.0.4.weight: copying a param with shape torch.Size([44, 512]) from checkpoint, the shape in current model is torch.Size([88, 512]).
size mismatch for mask_estimators.0.to_freqs.23.0.4.bias: copying a param with shape torch.Size([44]) from checkpoint, the shape in current model is torch.Size([88]).
size mismatch for mask_estimators.0.to_freqs.24.0.4.weight: copying a param with shape torch.Size([52, 512]) from checkpoint, the shape in current model is torch.Size([104, 512]).
size mismatch for mask_estimators.0.to_freqs.24.0.4.bias: copying a param with shape torch.Size([52]) from checkpoint, the shape in current model is torch.Size([104]).
size mismatch for mask_estimators.0.to_freqs.25.0.4.weight: copying a param with shape torch.Size([52, 512]) from checkpoint, the shape in current model is torch.Size([104, 512]).
size mismatch for mask_estimators.0.to_freqs.25.0.4.bias: copying a param with shape torch.Size([52]) from checkpoint, the shape in current model is torch.Size([104]).
size mismatch for mask_estimators.0.to_freqs.26.0.4.weight: copying a param with shape torch.Size([52, 512]) from checkpoint, the shape in current model is torch.Size([104, 512]).
size mismatch for mask_estimators.0.to_freqs.26.0.4.bias: copying a param with shape torch.Size([52]) from checkpoint, the shape in current model is torch.Size([104]).
size mismatch for mask_estimators.0.to_freqs.27.0.4.weight: copying a param with shape torch.Size([60, 512]) from checkpoint, the shape in current model is torch.Size([120, 512]).
size mismatch for mask_estimators.0.to_freqs.27.0.4.bias: copying a param with shape torch.Size([60]) from checkpoint, the shape in current model is torch.Size([120]).
size mismatch for mask_estimators.0.to_freqs.28.0.4.weight: copying a param with shape torch.Size([64, 512]) from checkpoint, the shape in current model is torch.Size([128, 512]).
size mismatch for mask_estimators.0.to_freqs.28.0.4.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for mask_estimators.0.to_freqs.29.0.4.weight: copying a param with shape torch.Size([68, 512]) from checkpoint, the shape in current model is torch.Size([136, 512]).
size mismatch for mask_estimators.0.to_freqs.29.0.4.bias: copying a param with shape torch.Size([68]) from checkpoint, the shape in current model is torch.Size([136]).
size mismatch for mask_estimators.0.to_freqs.30.0.4.weight: copying a param with shape torch.Size([76, 512]) from checkpoint, the shape in current model is torch.Size([152, 512]).
size mismatch for mask_estimators.0.to_freqs.30.0.4.bias: copying a param with shape torch.Size([76]) from checkpoint, the shape in current model is torch.Size([152]).
size mismatch for mask_estimators.0.to_freqs.31.0.4.weight: copying a param with shape torch.Size([80, 512]) from checkpoint, the shape in current model is torch.Size([160, 512]).
size mismatch for mask_estimators.0.to_freqs.31.0.4.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([160]).
size mismatch for mask_estimators.0.to_freqs.32.0.4.weight: copying a param with shape torch.Size([80, 512]) from checkpoint, the shape in current model is torch.Size([160, 512]).
size mismatch for mask_estimators.0.to_freqs.32.0.4.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([160]).
size mismatch for mask_estimators.0.to_freqs.33.0.4.weight: copying a param with shape torch.Size([88, 512]) from checkpoint, the shape in current model is torch.Size([176, 512]).
size mismatch for mask_estimators.0.to_freqs.33.0.4.bias: copying a param with shape torch.Size([88]) from checkpoint, the shape in current model is torch.Size([176]).
size mismatch for mask_estimators.0.to_freqs.34.0.4.weight: copying a param with shape torch.Size([96, 512]) from checkpoint, the shape in current model is torch.Size([192, 512]).
size mismatch for mask_estimators.0.to_freqs.34.0.4.bias: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([192]).
size mismatch for mask_estimators.0.to_freqs.35.0.4.weight: copying a param with shape torch.Size([104, 512]) from checkpoint, the shape in current model is torch.Size([208, 512]).
size mismatch for mask_estimators.0.to_freqs.35.0.4.bias: copying a param with shape torch.Size([104]) from checkpoint, the shape in current model is torch.Size([208]).
size mismatch for mask_estimators.0.to_freqs.36.0.4.weight: copying a param with shape torch.Size([112, 512]) from checkpoint, the shape in current model is torch.Size([224, 512]).
size mismatch for mask_estimators.0.to_freqs.36.0.4.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([224]).
size mismatch for mask_estimators.0.to_freqs.37.0.4.weight: copying a param with shape torch.Size([116, 512]) from checkpoint, the shape in current model is torch.Size([232, 512]).
size mismatch for mask_estimators.0.to_freqs.37.0.4.bias: copying a param with shape torch.Size([116]) from checkpoint, the shape in current model is torch.Size([232]).
size mismatch for mask_estimators.0.to_freqs.38.0.4.weight: copying a param with shape torch.Size([124, 512]) from checkpoint, the shape in current model is torch.Size([248, 512]).
size mismatch for mask_estimators.0.to_freqs.38.0.4.bias: copying a param with shape torch.Size([124]) from checkpoint, the shape in current model is torch.Size([248]).
size mismatch for mask_estimators.0.to_freqs.39.0.4.weight: copying a param with shape torch.Size([132, 512]) from checkpoint, the shape in current model is torch.Size([264, 512]).
size mismatch for mask_estimators.0.to_freqs.39.0.4.bias: copying a param with shape torch.Size([132]) from checkpoint, the shape in current model is torch.Size([264]).
size mismatch for mask_estimators.0.to_freqs.40.0.4.weight: copying a param with shape torch.Size([144, 512]) from checkpoint, the shape in current model is torch.Size([288, 512]).
size mismatch for mask_estimators.0.to_freqs.40.0.4.bias: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([288]).
size mismatch for mask_estimators.0.to_freqs.41.0.4.weight: copying a param with shape torch.Size([156, 512]) from checkpoint, the shape in current model is torch.Size([312, 512]).
size mismatch for mask_estimators.0.to_freqs.41.0.4.bias: copying a param with shape torch.Size([156]) from checkpoint, the shape in current model is torch.Size([312]).
size mismatch for mask_estimators.0.to_freqs.42.0.4.weight: copying a param with shape torch.Size([164, 512]) from checkpoint, the shape in current model is torch.Size([328, 512]).
size mismatch for mask_estimators.0.to_freqs.42.0.4.bias: copying a param with shape torch.Size([164]) from checkpoint, the shape in current model is torch.Size([328]).
size mismatch for mask_estimators.0.to_freqs.43.0.4.weight: copying a param with shape torch.Size([176, 512]) from checkpoint, the shape in current model is torch.Size([352, 512]).
size mismatch for mask_estimators.0.to_freqs.43.0.4.bias: copying a param with shape torch.Size([176]) from checkpoint, the shape in current model is torch.Size([352]).
size mismatch for mask_estimators.0.to_freqs.44.0.4.weight: copying a param with shape torch.Size([188, 512]) from checkpoint, the shape in current model is torch.Size([376, 512]).
size mismatch for mask_estimators.0.to_freqs.44.0.4.bias: copying a param with shape torch.Size([188]) from checkpoint, the shape in current model is torch.Size([376]).
size mismatch for mask_estimators.0.to_freqs.45.0.4.weight: copying a param with shape torch.Size([200, 512]) from checkpoint, the shape in current model is torch.Size([400, 512]).
size mismatch for mask_estimators.0.to_freqs.45.0.4.bias: copying a param with shape torch.Size([200]) from checkpoint, the shape in current model is torch.Size([400]).
size mismatch for mask_estimators.0.to_freqs.46.0.4.weight: copying a param with shape torch.Size([216, 512]) from checkpoint, the shape in current model is torch.Size([432, 512]).
size mismatch for mask_estimators.0.to_freqs.46.0.4.bias: copying a param with shape torch.Size([216]) from checkpoint, the shape in current model is torch.Size([432]).
size mismatch for mask_estimators.0.to_freqs.47.0.4.weight: copying a param with shape torch.Size([228, 512]) from checkpoint, the shape in current model is torch.Size([456, 512]).
size mismatch for mask_estimators.0.to_freqs.47.0.4.bias: copying a param with shape torch.Size([228]) from checkpoint, the shape in current model is torch.Size([456]).
size mismatch for mask_estimators.0.to_freqs.48.0.4.weight: copying a param with shape torch.Size([244, 512]) from checkpoint, the shape in current model is torch.Size([488, 512]).
size mismatch for mask_estimators.0.to_freqs.48.0.4.bias: copying a param with shape torch.Size([244]) from checkpoint, the shape in current model is torch.Size([488]).
size mismatch for mask_estimators.0.to_freqs.49.0.4.weight: copying a param with shape torch.Size([264, 512]) from checkpoint, the shape in current model is torch.Size([528, 512]).
size mismatch for mask_estimators.0.to_freqs.49.0.4.bias: copying a param with shape torch.Size([264]) from checkpoint, the shape in current model is torch.Size([528]).
size mismatch for mask_estimators.0.to_freqs.50.0.4.weight: copying a param with shape torch.Size([284, 512]) from checkpoint, the shape in current model is torch.Size([568, 512]).
size mismatch for mask_estimators.0.to_freqs.50.0.4.bias: copying a param with shape torch.Size([284]) from checkpoint, the shape in current model is torch.Size([568]).
size mismatch for mask_estimators.0.to_freqs.51.0.4.weight: copying a param with shape torch.Size([304, 512]) from checkpoint, the shape in current model is torch.Size([608, 512]).
size mismatch for mask_estimators.0.to_freqs.51.0.4.bias: copying a param with shape torch.Size([304]) from checkpoint, the shape in current model is torch.Size([608]).
size mismatch for mask_estimators.0.to_freqs.52.0.4.weight: copying a param with shape torch.Size([320, 512]) from checkpoint, the shape in current model is torch.Size([640, 512]).
size mismatch for mask_estimators.0.to_freqs.52.0.4.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([640]).
size mismatch for mask_estimators.0.to_freqs.53.0.4.weight: copying a param with shape torch.Size([344, 512]) from checkpoint, the shape in current model is torch.Size([688, 512]).
size mismatch for mask_estimators.0.to_freqs.53.0.4.bias: copying a param with shape torch.Size([344]) from checkpoint, the shape in current model is torch.Size([688]).
size mismatch for mask_estimators.0.to_freqs.54.0.4.weight: copying a param with shape torch.Size([372, 512]) from checkpoint, the shape in current model is torch.Size([744, 512]).
size mismatch for mask_estimators.0.to_freqs.54.0.4.bias: copying a param with shape torch.Size([372]) from checkpoint, the shape in current model is torch.Size([744]).
size mismatch for mask_estimators.0.to_freqs.55.0.4.weight: copying a param with shape torch.Size([396, 512]) from checkpoint, the shape in current model is torch.Size([792, 512]).
size mismatch for mask_estimators.0.to_freqs.55.0.4.bias: copying a param with shape torch.Size([396]) from checkpoint, the shape in current model is torch.Size([792]).
size mismatch for mask_estimators.0.to_freqs.56.0.4.weight: copying a param with shape torch.Size([420, 512]) from checkpoint, the shape in current model is torch.Size([840, 512]).
size mismatch for mask_estimators.0.to_freqs.56.0.4.bias: copying a param with shape torch.Size([420]) from checkpoint, the shape in current model is torch.Size([840]).
size mismatch for mask_estimators.0.to_freqs.57.0.4.weight: copying a param with shape torch.Size([452, 512]) from checkpoint, the shape in current model is torch.Size([904, 512]).
size mismatch for mask_estimators.0.to_freqs.57.0.4.bias: copying a param with shape torch.Size([452]) from checkpoint, the shape in current model is torch.Size([904]).
size mismatch for mask_estimators.0.to_freqs.58.0.4.weight: copying a param with shape torch.Size([488, 512]) from checkpoint, the shape in current model is torch.Size([976, 512]).
size mismatch for mask_estimators.0.to_freqs.58.0.4.bias: copying a param with shape torch.Size([488]) from checkpoint, the shape in current model is torch.Size([976]).
size mismatch for mask_estimators.0.to_freqs.59.0.4.weight: copying a param with shape torch.Size([520, 512]) from checkpoint, the shape in current model is torch.Size([1040, 512]).
size mismatch for mask_estimators.0.to_freqs.59.0.4.bias: copying a param with shape torch.Size([520]) from checkpoint, the shape in current model is torch.Size([1040]).
size mismatch for mask_estimators.1.to_freqs.0.0.4.weight: copying a param with shape torch.Size([28, 512]) from checkpoint, the shape in current model is torch.Size([56, 512]).
size mismatch for mask_estimators.1.to_freqs.0.0.4.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([56]).
size mismatch for mask_estimators.1.to_freqs.1.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.1.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.2.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.2.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.3.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.3.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.4.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.4.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.5.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.5.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.6.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.6.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.7.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.7.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.8.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.8.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.9.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.9.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.10.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.10.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.11.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.11.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.12.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.12.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.13.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.13.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.14.0.4.weight: copying a param with shape torch.Size([24, 512]) from checkpoint, the shape in current model is torch.Size([48, 512]).
size mismatch for mask_estimators.1.to_freqs.14.0.4.bias: copying a param with shape torch.Size([24]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for mask_estimators.1.to_freqs.15.0.4.weight: copying a param with shape torch.Size([28, 512]) from checkpoint, the shape in current model is torch.Size([56, 512]).
size mismatch for mask_estimators.1.to_freqs.15.0.4.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([56]).
size mismatch for mask_estimators.1.to_freqs.16.0.4.weight: copying a param with shape torch.Size([28, 512]) from checkpoint, the shape in current model is torch.Size([56, 512]).
size mismatch for mask_estimators.1.to_freqs.16.0.4.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([56]).
size mismatch for mask_estimators.1.to_freqs.17.0.4.weight: copying a param with shape torch.Size([28, 512]) from checkpoint, the shape in current model is torch.Size([56, 512]).
size mismatch for mask_estimators.1.to_freqs.17.0.4.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([56]).
size mismatch for mask_estimators.1.to_freqs.18.0.4.weight: copying a param with shape torch.Size([36, 512]) from checkpoint, the shape in current model is torch.Size([72, 512]).
size mismatch for mask_estimators.1.to_freqs.18.0.4.bias: copying a param with shape torch.Size([36]) from checkpoint, the shape in current model is torch.Size([72]).
size mismatch for mask_estimators.1.to_freqs.19.0.4.weight: copying a param with shape torch.Size([36, 512]) from checkpoint, the shape in current model is torch.Size([72, 512]).
size mismatch for mask_estimators.1.to_freqs.19.0.4.bias: copying a param with shape torch.Size([36]) from checkpoint, the shape in current model is torch.Size([72]).
size mismatch for mask_estimators.1.to_freqs.20.0.4.weight: copying a param with shape torch.Size([36, 512]) from checkpoint, the shape in current model is torch.Size([72, 512]).
size mismatch for mask_estimators.1.to_freqs.20.0.4.bias: copying a param with shape torch.Size([36]) from checkpoint, the shape in current model is torch.Size([72]).
size mismatch for mask_estimators.1.to_freqs.21.0.4.weight: copying a param with shape torch.Size([40, 512]) from checkpoint, the shape in current model is torch.Size([80, 512]).
size mismatch for mask_estimators.1.to_freqs.21.0.4.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([80]).
size mismatch for mask_estimators.1.to_freqs.22.0.4.weight: copying a param with shape torch.Size([40, 512]) from checkpoint, the shape in current model is torch.Size([80, 512]).
size mismatch for mask_estimators.1.to_freqs.22.0.4.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([80]).
size mismatch for mask_estimators.1.to_freqs.23.0.4.weight: copying a param with shape torch.Size([44, 512]) from checkpoint, the shape in current model is torch.Size([88, 512]).
size mismatch for mask_estimators.1.to_freqs.23.0.4.bias: copying a param with shape torch.Size([44]) from checkpoint, the shape in current model is torch.Size([88]).
size mismatch for mask_estimators.1.to_freqs.24.0.4.weight: copying a param with shape torch.Size([52, 512]) from checkpoint, the shape in current model is torch.Size([104, 512]).
size mismatch for mask_estimators.1.to_freqs.24.0.4.bias: copying a param with shape torch.Size([52]) from checkpoint, the shape in current model is torch.Size([104]).
size mismatch for mask_estimators.1.to_freqs.25.0.4.weight: copying a param with shape torch.Size([52, 512]) from checkpoint, the shape in current model is torch.Size([104, 512]).
size mismatch for mask_estimators.1.to_freqs.25.0.4.bias: copying a param with shape torch.Size([52]) from checkpoint, the shape in current model is torch.Size([104]).
size mismatch for mask_estimators.1.to_freqs.26.0.4.weight: copying a param with shape torch.Size([52, 512]) from checkpoint, the shape in current model is torch.Size([104, 512]).
size mismatch for mask_estimators.1.to_freqs.26.0.4.bias: copying a param with shape torch.Size([52]) from checkpoint, the shape in current model is torch.Size([104]).
size mismatch for mask_estimators.1.to_freqs.27.0.4.weight: copying a param with shape torch.Size([60, 512]) from checkpoint, the shape in current model is torch.Size([120, 512]).
size mismatch for mask_estimators.1.to_freqs.27.0.4.bias: copying a param with shape torch.Size([60]) from checkpoint, the shape in current model is torch.Size([120]).
size mismatch for mask_estimators.1.to_freqs.28.0.4.weight: copying a param with shape torch.Size([64, 512]) from checkpoint, the shape in current model is torch.Size([128, 512]).
size mismatch for mask_estimators.1.to_freqs.28.0.4.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for mask_estimators.1.to_freqs.29.0.4.weight: copying a param with shape torch.Size([68, 512]) from checkpoint, the shape in current model is torch.Size([136, 512]).
size mismatch for mask_estimators.1.to_freqs.29.0.4.bias: copying a param with shape torch.Size([68]) from checkpoint, the shape in current model is torch.Size([136]).
size mismatch for mask_estimators.1.to_freqs.30.0.4.weight: copying a param with shape torch.Size([76, 512]) from checkpoint, the shape in current model is torch.Size([152, 512]).
size mismatch for mask_estimators.1.to_freqs.30.0.4.bias: copying a param with shape torch.Size([76]) from checkpoint, the shape in current model is torch.Size([152]).
size mismatch for mask_estimators.1.to_freqs.31.0.4.weight: copying a param with shape torch.Size([80, 512]) from checkpoint, the shape in current model is torch.Size([160, 512]).
size mismatch for mask_estimators.1.to_freqs.31.0.4.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([160]).
size mismatch for mask_estimators.1.to_freqs.32.0.4.weight: copying a param with shape torch.Size([80, 512]) from checkpoint, the shape in current model is torch.Size([160, 512]).
size mismatch for mask_estimators.1.to_freqs.32.0.4.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([160]).
size mismatch for mask_estimators.1.to_freqs.33.0.4.weight: copying a param with shape torch.Size([88, 512]) from checkpoint, the shape in current model is torch.Size([176, 512]).
size mismatch for mask_estimators.1.to_freqs.33.0.4.bias: copying a param with shape torch.Size([88]) from checkpoint, the shape in current model is torch.Size([176]).
size mismatch for mask_estimators.1.to_freqs.34.0.4.weight: copying a param with shape torch.Size([96, 512]) from checkpoint, the shape in current model is torch.Size([192, 512]).
size mismatch for mask_estimators.1.to_freqs.34.0.4.bias: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([192]).
size mismatch for mask_estimators.1.to_freqs.35.0.4.weight: copying a param with shape torch.Size([104, 512]) from checkpoint, the shape in current model is torch.Size([208, 512]).
size mismatch for mask_estimators.1.to_freqs.35.0.4.bias: copying a param with shape torch.Size([104]) from checkpoint, the shape in current model is torch.Size([208]).
size mismatch for mask_estimators.1.to_freqs.36.0.4.weight: copying a param with shape torch.Size([112, 512]) from checkpoint, the shape in current model is torch.Size([224, 512]).
size mismatch for mask_estimators.1.to_freqs.36.0.4.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([224]).
size mismatch for mask_estimators.1.to_freqs.37.0.4.weight: copying a param with shape torch.Size([116, 512]) from checkpoint, the shape in current model is torch.Size([232, 512]).
size mismatch for mask_estimators.1.to_freqs.37.0.4.bias: copying a param with shape torch.Size([116]) from checkpoint, the shape in current model is torch.Size([232]).
size mismatch for mask_estimators.1.to_freqs.38.0.4.weight: copying a param with shape torch.Size([124, 512]) from checkpoint, the shape in current model is torch.Size([248, 512]).
size mismatch for mask_estimators.1.to_freqs.38.0.4.bias: copying a param with shape torch.Size([124]) from checkpoint, the shape in current model is torch.Size([248]).
size mismatch for mask_estimators.1.to_freqs.39.0.4.weight: copying a param with shape torch.Size([132, 512]) from checkpoint, the shape in current model is torch.Size([264, 512]).
size mismatch for mask_estimators.1.to_freqs.39.0.4.bias: copying a param with shape torch.Size([132]) from checkpoint, the shape in current model is torch.Size([264]).
size mismatch for mask_estimators.1.to_freqs.40.0.4.weight: copying a param with shape torch.Size([144, 512]) from checkpoint, the shape in current model is torch.Size([288, 512]).
size mismatch for mask_estimators.1.to_freqs.40.0.4.bias: copying a param with shape torch.Size([144]) from checkpoint, the shape in current model is torch.Size([288]).
size mismatch for mask_estimators.1.to_freqs.41.0.4.weight: copying a param with shape torch.Size([156, 512]) from checkpoint, the shape in current model is torch.Size([312, 512]).
size mismatch for mask_estimators.1.to_freqs.41.0.4.bias: copying a param with shape torch.Size([156]) from checkpoint, the shape in current model is torch.Size([312]).
size mismatch for mask_estimators.1.to_freqs.42.0.4.weight: copying a param with shape torch.Size([164, 512]) from checkpoint, the shape in current model is torch.Size([328, 512]).
size mismatch for mask_estimators.1.to_freqs.42.0.4.bias: copying a param with shape torch.Size([164]) from checkpoint, the shape in current model is torch.Size([328]).
size mismatch for mask_estimators.1.to_freqs.43.0.4.weight: copying a param with shape torch.Size([176, 512]) from checkpoint, the shape in current model is torch.Size([352, 512]).
size mismatch for mask_estimators.1.to_freqs.43.0.4.bias: copying a param with shape torch.Size([176]) from checkpoint, the shape in current model is torch.Size([352]).
size mismatch for mask_estimators.1.to_freqs.44.0.4.weight: copying a param with shape torch.Size([188, 512]) from checkpoint, the shape in current model is torch.Size([376, 512]).
size mismatch for mask_estimators.1.to_freqs.44.0.4.bias: copying a param with shape torch.Size([188]) from checkpoint, the shape in current model is torch.Size([376]).
size mismatch for mask_estimators.1.to_freqs.45.0.4.weight: copying a param with shape torch.Size([200, 512]) from checkpoint, the shape in current model is torch.Size([400, 512]).
size mismatch for mask_estimators.1.to_freqs.45.0.4.bias: copying a param with shape torch.Size([200]) from checkpoint, the shape in current model is torch.Size([400]).
size mismatch for mask_estimators.1.to_freqs.46.0.4.weight: copying a param with shape torch.Size([216, 512]) from checkpoint, the shape in current model is torch.Size([432, 512]).
size mismatch for mask_estimators.1.to_freqs.46.0.4.bias: copying a param with shape torch.Size([216]) from checkpoint, the shape in current model is torch.Size([432]).
size mismatch for mask_estimators.1.to_freqs.47.0.4.weight: copying a param with shape torch.Size([228, 512]) from checkpoint, the shape in current model is torch.Size([456, 512]).
size mismatch for mask_estimators.1.to_freqs.47.0.4.bias: copying a param with shape torch.Size([228]) from checkpoint, the shape in current model is torch.Size([456]).
size mismatch for mask_estimators.1.to_freqs.48.0.4.weight: copying a param with shape torch.Size([244, 512]) from checkpoint, the shape in current model is torch.Size([488, 512]).
size mismatch for mask_estimators.1.to_freqs.48.0.4.bias: copying a param with shape torch.Size([244]) from checkpoint, the shape in current model is torch.Size([488]).
size mismatch for mask_estimators.1.to_freqs.49.0.4.weight: copying a param with shape torch.Size([264, 512]) from checkpoint, the shape in current model is torch.Size([528, 512]).
size mismatch for mask_estimators.1.to_freqs.49.0.4.bias: copying a param with shape torch.Size([264]) from checkpoint, the shape in current model is torch.Size([528]).
size mismatch for mask_estimators.1.to_freqs.50.0.4.weight: copying a param with shape torch.Size([284, 512]) from checkpoint, the shape in current model is torch.Size([568, 512]).
size mismatch for mask_estimators.1.to_freqs.50.0.4.bias: copying a param with shape torch.Size([284]) from checkpoint, the shape in current model is torch.Size([568]).
size mismatch for mask_estimators.1.to_freqs.51.0.4.weight: copying a param with shape torch.Size([304, 512]) from checkpoint, the shape in current model is torch.Size([608, 512]).
size mismatch for mask_estimators.1.to_freqs.51.0.4.bias: copying a param with shape torch.Size([304]) from checkpoint, the shape in current model is torch.Size([608]).
size mismatch for mask_estimators.1.to_freqs.52.0.4.weight: copying a param with shape torch.Size([320, 512]) from checkpoint, the shape in current model is torch.Size([640, 512]).
size mismatch for mask_estimators.1.to_freqs.52.0.4.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([640]).
size mismatch for mask_estimators.1.to_freqs.53.0.4.weight: copying a param with shape torch.Size([344, 512]) from checkpoint, the shape in current model is torch.Size([688, 512]).
size mismatch for mask_estimators.1.to_freqs.53.0.4.bias: copying a param with shape torch.Size([344]) from checkpoint, the shape in current model is torch.Size([688]).
size mismatch for mask_estimators.1.to_freqs.54.0.4.weight: copying a param with shape torch.Size([372, 512]) from checkpoint, the shape in current model is torch.Size([744, 512]).
size mismatch for mask_estimators.1.to_freqs.54.0.4.bias: copying a param with shape torch.Size([372]) from checkpoint, the shape in current model is torch.Size([744]).
size mismatch for mask_estimators.1.to_freqs.55.0.4.weight: copying a param with shape torch.Size([396, 512]) from checkpoint, the shape in current model is torch.Size([792, 512]).
size mismatch for mask_estimators.1.to_freqs.55.0.4.bias: copying a param with shape torch.Size([396]) from checkpoint, the shape in current model is torch.Size([792]).
size mismatch for mask_estimators.1.to_freqs.56.0.4.weight: copying a param with shape torch.Size([420, 512]) from checkpoint, the shape in current model is torch.Size([840, 512]).
size mismatch for mask_estimators.1.to_freqs.56.0.4.bias: copying a param with shape torch.Size([420]) from checkpoint, the shape in current model is torch.Size([840]).
size mismatch for mask_estimators.1.to_freqs.57.0.4.weight: copying a param with shape torch.Size([452, 512]) from checkpoint, the shape in current model is torch.Size([904, 512]).
size mismatch for mask_estimators.1.to_freqs.57.0.4.bias: copying a param with shape torch.Size([452]) from checkpoint, the shape in current model is torch.Size([904]).
size mismatch for mask_estimators.1.to_freqs.58.0.4.weight: copying a param with shape torch.Size([488, 512]) from checkpoint, the shape in current model is torch.Size([976, 512]).
size mismatch for mask_estimators.1.to_freqs.58.0.4.bias: copying a param with shape torch.Size([488]) from checkpoint, the shape in current model is torch.Size([976]).
size mismatch for mask_estimators.1.to_freqs.59.0.4.weight: copying a param with shape torch.Size([520, 512]) from checkpoint, the shape in current model is torch.Size([1040, 512]).
size mismatch for mask_estimators.1.to_freqs.59.0.4.bias: copying a param with shape torch.Size([520]) from checkpoint, the shape in current model is torch.Size([1040]).
Process failed with return code 1

please convert your input audio to mono.

I just converted the file to mono, and the following message appears.

File "C:\Program Files\Music Source Separation Training 1.0.12\inference.py", line 184, in
proc_folder(None)
File "C:\Program Files\Music Source Separation Training 1.0.12\inference.py", line 180, in proc_folder
run_folder(model, args, config, device, verbose=True)
File "C:\Program Files\Music Source Separation Training 1.0.12\inference.py", line 79, in run_folder
waveforms = demix(config, model, mix, device, pbar=detailed_pbar, model_type=args.model_type)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Program Files\Music Source Separation Training 1.0.12\utils.py", line 408, in demix
return demix_track(config, model, mix, device, pbar=pbar)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Program Files\Music Source Separation Training 1.0.12\utils.py", line 134, in demix_track
x = model(arr)
^^^^^^^^^^
File "C:\Users\Kveris\AppData\Local\Programs\Python\Python311\Lib\site-packages\torch\nn\modules\module.py", line 1779, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\Kveris\AppData\Local\Programs\Python\Python311\Lib\site-packages\torch\nn\modules\module.py", line 1790, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Program Files\Music Source Separation Training 1.0.12\models\bs_roformer\mel_band_roformer.py", line 566, in forward
self.stereo and channels == 2), 'stereo needs to be set to True if passing in audio signal that is stereo (channel dimension of 2). also need to be False if mono (channel dimension of 1)'
^^^^^^^^^^^^^
AssertionError: stereo needs to be set to True if passing in audio signal that is stereo (channel dimension of 2). also need to be False if mono (channel dimension of 1)

you should probably contact the creator of this program, i am not responsible for this error.
I only make models, i tested it on both MSST and MSSTWebui and it works fine.

gilliaan locked this discussion

Sign up or log in to comment