updated imports
Browse files- model_Small.py +0 -1
model_Small.py
CHANGED
|
@@ -5,7 +5,6 @@ import warnings
|
|
| 5 |
warnings.filterwarnings("ignore")
|
| 6 |
import torchvision
|
| 7 |
import parameters as params
|
| 8 |
-
import timm
|
| 9 |
|
| 10 |
class DinoV2FeatureExtractor(nn.Module):
|
| 11 |
def __init__(self, out_channels=256, out_size=(64, 64)):
|
|
|
|
| 5 |
warnings.filterwarnings("ignore")
|
| 6 |
import torchvision
|
| 7 |
import parameters as params
|
|
|
|
| 8 |
|
| 9 |
class DinoV2FeatureExtractor(nn.Module):
|
| 10 |
def __init__(self, out_channels=256, out_size=(64, 64)):
|