rishikksh20 / hifigan-denoiser

HiFi-GAN: High Fidelity Denoising and Dereverberation Based on Speech Deep Features in Adversarial Networks
Apache License 2.0
208 stars 45 forks source link

KeyError: '__getstate__' #4

Open KevinBaylor opened 3 years ago

KevinBaylor commented 3 years ago

Hi, thanks for opensourcing your code! During the training process, I met an error with the command bellow.

COMMAND python train.py -c config.yaml

ERROR Initializing Training Process.. Batch size per GPU : 0 Traceback (most recent call last): File "train.py", line 304, in main() File "train.py", line 298, in main mp.spawn(train, nprocs=hp.train.num_gpus, args=(args, hp, hp_str,)) File "/home/lian/.conda/envs/hifi-GAN-denoise/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 199, in spawn return start_processes(fn, args, nprocs, join, daemon, start_method='spawn') File "/home/lian/.conda/envs/hifi-GAN-denoise/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 148, in start_processes process.start() File "/home/lian/.conda/envs/hifi-GAN-denoise/lib/python3.6/multiprocessing/process.py", line 105, in start self._popen = self._Popen(self) File "/home/lian/.conda/envs/hifi-GAN-denoise/lib/python3.6/multiprocessing/context.py", line 284, in _Popen return Popen(process_obj) File "/home/lian/.conda/envs/hifi-GAN-denoise/lib/python3.6/multiprocessing/popen_spawn_posix.py", line 32, in init super().init(process_obj) File "/home/lian/.conda/envs/hifi-GAN-denoise/lib/python3.6/multiprocessing/popen_fork.py", line 19, in init self._launch(process_obj) File "/home/lian/.conda/envs/hifi-GAN-denoise/lib/python3.6/multiprocessing/popen_spawn_posix.py", line 47, in _launch reduction.dump(process_obj, fp) File "/home/lian/.conda/envs/hifi-GAN-denoise/lib/python3.6/multiprocessing/reduction.py", line 60, in dump ForkingPickler(file, protocol).dump(obj) KeyError: 'getstate'

Would you like to tell me why it happened and how to solve it? Thank you! Have a nice day.

skol101 commented 2 years ago

Yep, doesn't work for multi gpu training.