Code release for ConvNeXt model
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
collinmccarthy opened this issue 2 years ago · comments
Hi,
Why do you need to call torch.cuda.synchronize() during train_one_epoch?
torch.cuda.synchronize()
Thanks, -Collin