Skip to content
Snippets Groups Projects
Commit da197382 authored by Engin Eren's avatar Engin Eren
Browse files

remove set_epoch :S

parent a116159f
No related branches found
No related tags found
1 merge request!3Test
Checking pipeline status
Hide whitespace changes
Inline Side-by-side
Showing
with 0 additions and 2 deletions
wganHCAL.py
+ 0
− 2
View file @ da197382
...@@ -306,8 +306,6 @@ def run(args): ...@@ -306,8 +306,6 @@ def run(args):
if args.world_size > 1: if args.world_size > 1:
train_loader.sampler.set_epoch(epoch) train_loader.sampler.set_epoch(epoch)
else:
train_loader.set_epoch(epoch)
train(args, mCrit, mGen, device, train_loader, optimizerD, optimizerG, epoch, experiment) train(args, mCrit, mGen, device, train_loader, optimizerD, optimizerG, epoch, experiment)
if args.rank == 0: if args.rank == 0:
......
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