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
Hi, In training the FlowNetC, the correlation layer runs so slow, It's about 5.5s a batch (batch size = 8) on TiTanX, I don't know whether there are some problems, and could you give me some advice for fast training? Thanks!
The text was updated successfully, but these errors were encountered:
@sczhou the issue was that non optimal number of threads per block were used for the backward function of the correlation layer, hence the slow speed.
it is now faster by about 5x.
Hi, In training the FlowNetC, the correlation layer runs so slow, It's about 5.5s a batch (batch size = 8) on TiTanX, I don't know whether there are some problems, and could you give me some advice for fast training? Thanks!
The text was updated successfully, but these errors were encountered: