try gtp vit
This commit is contained in:
parent
41a5c7b05a
commit
23c44d3582
@ -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.layers import DropPath, trunc_normal_
|
||||||
from timm.models.registry import register_model
|
from timm.models.registry import register_model
|
||||||
from timm.layers.helpers import to_2tuple
|
from timm.layers.helpers import to_2tuple
|
||||||
|
from typing import *
|
||||||
|
|
||||||
|
|
||||||
class ConvE(torch.nn.Module):
|
class ConvE(torch.nn.Module):
|
||||||
|
Loading…
Reference in New Issue
Block a user