try gtp vit

2024-04-28 11:59:09 +07:00
parent 41a5c7b05a
commit 23c44d3582
1 changed files with 1 additions and 0 deletions
--- a/models.py
+++ b/models.py
@@ -10,6 +10,7 @@ from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
 from timm.models.layers import DropPath, trunc_normal_
 from timm.models.registry import register_model
 from timm.layers.helpers import to_2tuple
+from typing import *


 class ConvE(torch.nn.Module):