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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -99,7 +99,7 @@ def main():
n_gpus = 1 n_gpus = 1
if n_gpus < 1: if n_gpus < 1:
# patch to unblock people without gpus. there is probably a better way. # 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 n_gpus = 1
os.environ["MASTER_ADDR"] = "localhost" os.environ["MASTER_ADDR"] = "localhost"
os.environ["MASTER_PORT"] = str(randint(20000, 55555)) os.environ["MASTER_PORT"] = str(randint(20000, 55555))