Building Vision Models upon Heat Conduction

Figure 1 . Throughput / GPU memory / FLOPs comparisons of our proposed approach (vHeat) with under different image resolutions. The throughput and GPU memory are tested on 80 GB Tesla A100 GPUs with batch size 64. Swin-B is tested with scaled window size here.