Update train.py

This commit is contained in:
RVC-Boss
2023-10-16 23:44:41 +08:00
committed by GitHub
parent 0472bc5283
commit 3cc0964600

View File

@@ -99,7 +99,7 @@ def main():
n_gpus = 1
if n_gpus < 1:
# patch to unblock people without gpus. there is probably a better way.
logger.warning("NO GPU DETECTED: falling back to CPU - this may take a while")
print("NO GPU DETECTED: falling back to CPU - this may take a while")
n_gpus = 1
os.environ["MASTER_ADDR"] = "localhost"
os.environ["MASTER_PORT"] = str(randint(20000, 55555))