Closed tengshaofeng closed 2 years ago
when I test with 256x256, it takes 4s with cpu, and if I test with 512x512, it takes 25s in cpu.
Try GPU mate and best place to start will be colab @tengshaofeng
when I test with 256x256, it takes 4s with cpu, and if I test with 512x512, it takes 25s in cpu.