Guanyuansheng / TFGAN-PLC

A Temporal-Spectral Generative Adversarial Network based End-to-end Packet Loss Concealment for Wideband Speech Transmission
30 stars 11 forks source link

Requirements of env and CUDA #12

Open XDHDD opened 1 year ago

XDHDD commented 1 year ago

Is any full list of requirements of packages and libraries for running the training ? Also trainer.py 22 string pesq isn't imported anywhere. After was able to run training on cpu I can't do the same using GPU. Two problems: "RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu" or "Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method"