mirror of
https://github.com/karpathy/nanochat.git
synced 2026-01-09 21:22:41 +00:00
Merge 41bd7b16f2 into 190d9515d0
This commit is contained in:
commit
6a31d9450b
|
|
@ -76,8 +76,7 @@ def print_banner():
|
|||
print0(banner)
|
||||
|
||||
def is_ddp():
|
||||
# TODO is there a proper way
|
||||
return int(os.environ.get('RANK', -1)) != -1
|
||||
return dist.is_available() and dist.is_initialized()
|
||||
|
||||
def get_dist_info():
|
||||
if is_ddp():
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user