RepLKNet 学习笔记

目录

原版测试代码：

torch版本：

RepLKNet 哔哩哔哩

---

原版测试代码：

1060显卡，224*224，平均时间10ms以内。

import time

import torch

from unireplknet import unireplknet_a

if __name__ == '__main__':
    net = unireplknet_a()

    net.cuda()
    net.eval()


    for i in range(20):