Open collinmccarthy opened 2 years ago
Hi,
Why do you need to call torch.cuda.synchronize() during train_one_epoch?
torch.cuda.synchronize()
Thanks, -Collin
Hi,
Why do you need to call
torch.cuda.synchronize()
during train_one_epoch?Thanks, -Collin