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

cleaning

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