This script does not work on 15 computers, with different versions of Python, the same goes for Google Colab
C:\Users\aboba\AppData\Local\Programs\Python\Python312\Lib\site-packages\torch\optim\lr_scheduler.py:143: UserWarning: Detected call of lr_scheduler.step() before optimizer.step(). In PyTorch 1.1.0 and later, you should call them in the opposite order: optimizer.step() before lr_scheduler.step(). Failure to do this will result in PyTorch skipping the first value of the learning rate schedule. See more details at https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate
warnings.warn("Detected call of lr_scheduler.step() before optimizer.step(). "
Downloading: "https://github.com/tarepan/SpeechMOS/zipball/v1.0.0" to C:\Users\aboba/.cache\torch\hub\v1.0.0.zip
Downloading: "https://github.com/tarepan/SpeechMOS/releases/download/v1.0.0/utmos22_strong_step7459_v1.pt" to C:\Users\aboba/.cache\torch\hub\checkpoints\utmos22_strong_step7459_v1.pt
100%|βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ| 392M/392M [00:45<00:00, 8.96MB/s]
C:\Users\aboba\AppData\Local\Programs\Python\Python312\Lib\site-packages\torch\nn\utils\weight_norm.py:28: UserWarning: torch.nn.utils.weight_norm is deprecated in favor of torch.nn.utils.parametrizations.weight_norm.
warnings.warn("torch.nn.utils.weight_norm is deprecated in favor of torch.nn.utils.parametrizations.weight_norm.")
0%| | 0/20000 [00:00<?, ?it/s]Traceback (most recent call last):
File "", line 1, in
File "C:\Users\aboba\AppData\Local\Programs\Python\Python312\Lib\multiprocessing\spawn.py", line 122, in spawn_main
exitcode = _main(fd, parent_sentinel)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\aboba\AppData\Local\Programs\Python\Python312\Lib\multiprocessing\spawn.py", line 132, in _main
self = reduction.pickle.load(from_parent)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: Can't get attribute 'WavDataset' on <module '__main__' (<class '_frozen_importlib.BuiltinImporter'>)>
Don't work, running the original script without changes
Wow thanks, I just needed to add __main__.py
, otherwise I was getting the WavDataset error every time on every version