Commit 8acddae7 authored by Anthony Larcher's avatar Anthony Larcher
Browse files


parent 383393c6
......@@ -1224,7 +1224,7 @@ def xtrain(dataset_description,
local_rank = -1
if "RANK" in os.environ:
local_rank = os.environ['RANK']
local_rank = int(os.environ['RANK'])
# Test to optimize
Supports Markdown
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