You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Yes right. Self attention module takes a lot of gpu memory. Increase the number of gpus for higher batch size. (or smaller image resolution, smaller number of attention modules, etc)
Batch_size = 6 ?
Right?
The text was updated successfully, but these errors were encountered: