Unverified Commit bf6ba258 authored by Tobias Fischer's avatar Tobias Fischer Committed by GitHub
Browse files

add world_size in if clause in load_from_http (#1396)

* add world_size in if clause

* add explanation

* remove LOCAL_RANK check
parent 990d8b6a
......@@ -278,7 +278,6 @@ def load_from_http(filename, map_location=None, model_dir=None):
dict or OrderedDict: The loaded checkpoint.
"""
rank, world_size = get_dist_info()
rank = int(os.environ.get('LOCAL_RANK', rank))
if rank == 0:
checkpoint = load_url(
filename, model_dir=model_dir, map_location=map_location)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment