Update train.py
This commit is contained in:
parent
08b22f036e
commit
c8b6be0a53
@ -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))
|
||||||
|
Loading…
x
Reference in New Issue
Block a user