Cyanogenoid / pytorch-vqa

Strong baseline for visual question answering
238 stars 97 forks source link

update for newer version of torch #26

Closed RedContritio closed 1 year ago

RedContritio commented 1 year ago
  1. use torch.no_grad instead of volatile
  2. use non_blocking arg instead of async