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

cleaning

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
torch.autograd.profiler.emit_nvtx(enabled=False)
......
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