Skip to content

GitLab

Commit 789afd89 authored by Christian Sarofeen's avatar Christian Sarofeen
Browse files

Need to set_epoch in distributed example for sampler

parent 3ab105d9
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
examples/distributed/main.py
View file @ 789afd89
...@@ -196,5 +196,9 @@ def test(): ...@@ -196,5 +196,9 @@ def test():
for epoch in range(1, args.epochs + 1): for epoch in range(1, args.epochs + 1):
#=====START: ADDED FOR DISTRIBUTED======
train_sampler.set_epoch(epoch)
#=====END: ADDED FOR DISTRIBUTED======
train(epoch) train(epoch)
test() test()
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