Update train.py

This commit is contained in:
RVC-Boss 2023-10-16 23:45:00 +08:00 committed by GitHub
parent 08b22f036e
commit c8b6be0a53
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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))