v1.0.4

model.properties

@@ -5,6 +5,6 @@ modelName=repvit-optimize_pytorch
 # 模型描述
 modelDescription=RepViT在iPhone 12上以1ms的延迟实现了超过80%的top-1准确率，本算法基于RepViT进一步优化后准确率超过82%。
 # 应用场景
-appScenario=训练,图像分类,制造,电商,医疗,能源,教育
+appScenario=训练,制造,电商,医疗,能源,教育
 # 框架类型
 frameType=pytorch

model/repvit.py

 import torch.nn as nn
 import torch.nn.functional as F
-from .triplet_attention import *
 
 def _make_divisible(v, divisor, min_value=None):
     """