tlkvstepan

Hello, Why it takes so much memory for the inference? The bottleneck I guess is cost a array at the beginning of contraction part of size h/4*w/4*d/4*64.. it should be...