mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Add missing imports

Browse Source
This commit is contained in:
Adam J. Stewart 2025-01-11 15:13:16 +01:00
parent 19aaea3c8f
commit f5c4d5cbb7
No known key found for this signature in database
GPG Key ID: C66C0675661156FC
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/layers/attention2d.py
View File

@ -1,4 +1,4 @@
from typing import List, Optional, Union from typing import List, Optional, Type, Union
import torch import torch
from torch import nn as nn from torch import nn as nn

2
timm/models/fastvit.py
View File

@ -7,7 +7,7 @@
# #
import os import os
from functools import partial from functools import partial
from typing import List, Optional, Tuple, Union from typing import List, Optional, Tuple, Type, Union
import torch import torch
import torch.nn as nn import torch.nn as nn

2
timm/models/swin_transformer_v2.py
View File

@ -13,7 +13,7 @@ Modifications and additions for timm hacked together by / Copyright 2022, Ross W
# Written by Ze Liu # Written by Ze Liu
# -------------------------------------------------------- # --------------------------------------------------------
import math import math
from typing import Callable, List, Optional, Tuple, Union from typing import Callable, List, Optional, Tuple, Type, Union
import torch import torch
import torch.nn as nn import torch.nn as nn

2
timm/models/vgg.py
View File

@ -5,7 +5,7 @@ timm functionality.
Copyright 2021 Ross Wightman Copyright 2021 Ross Wightman
""" """
from typing import Any, Dict, List, Optional, Union, cast from typing import Any, Dict, List, Optional, Type, Union, cast
import torch import torch
import torch.nn as nn import torch.nn as nn