Initial commit
Browse files- spectre/__init__.py +0 -3
- spectre/models/__init__.py +0 -10
- spectre/utils/__init__.py +0 -22
spectre/__init__.py
CHANGED
|
@@ -19,9 +19,6 @@ __all__ = [
|
|
| 19 |
"SpectreImageFeatureExtractor",
|
| 20 |
"MODEL_CONFIGS",
|
| 21 |
"models",
|
| 22 |
-
"data",
|
| 23 |
-
"transforms",
|
| 24 |
-
"ssl",
|
| 25 |
"utils",
|
| 26 |
"__version__",
|
| 27 |
"__author__",
|
|
|
|
| 19 |
"SpectreImageFeatureExtractor",
|
| 20 |
"MODEL_CONFIGS",
|
| 21 |
"models",
|
|
|
|
|
|
|
|
|
|
| 22 |
"utils",
|
| 23 |
"__version__",
|
| 24 |
"__author__",
|
spectre/models/__init__.py
CHANGED
|
@@ -28,14 +28,4 @@ __all__ = [
|
|
| 28 |
'feat_vit_small',
|
| 29 |
'feat_vit_base',
|
| 30 |
'feat_vit_large',
|
| 31 |
-
'ResNet',
|
| 32 |
-
'resnet18',
|
| 33 |
-
'resnet34',
|
| 34 |
-
'resnet50',
|
| 35 |
-
'resnet101',
|
| 36 |
-
'resnext50',
|
| 37 |
-
'resnext101',
|
| 38 |
-
'EoMT',
|
| 39 |
-
'UPA',
|
| 40 |
-
'SEoMT',
|
| 41 |
]
|
|
|
|
| 28 |
'feat_vit_small',
|
| 29 |
'feat_vit_base',
|
| 30 |
'feat_vit_large',
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
]
|
spectre/utils/__init__.py
CHANGED
|
@@ -39,23 +39,6 @@ __all__ = [
|
|
| 39 |
"to_2tuple",
|
| 40 |
"to_3tuple",
|
| 41 |
"to_4tuple",
|
| 42 |
-
"save_state",
|
| 43 |
-
"load_state",
|
| 44 |
-
"extract_model_from_checkpoint_dinov2",
|
| 45 |
-
"extract_model_from_checkpoint_siglip",
|
| 46 |
-
"extended_collate_dino",
|
| 47 |
-
"extended_collate_siglip",
|
| 48 |
-
"collate_add_filenames",
|
| 49 |
-
"setup",
|
| 50 |
-
"get_dataloader",
|
| 51 |
-
"is_enabled",
|
| 52 |
-
"get_global_size",
|
| 53 |
-
"get_global_rank",
|
| 54 |
-
"get_local_size",
|
| 55 |
-
"get_local_rank",
|
| 56 |
-
"init_distributed",
|
| 57 |
-
"add_lora_adapters",
|
| 58 |
-
"random_block_mask",
|
| 59 |
"deactivate_requires_grad_and_to_eval",
|
| 60 |
"activate_requires_grad_and_to_train",
|
| 61 |
"update_momentum",
|
|
@@ -79,9 +62,4 @@ __all__ = [
|
|
| 79 |
"Format",
|
| 80 |
"nchwd_to",
|
| 81 |
"nhwdc_to",
|
| 82 |
-
"get_param_groups_with_decay",
|
| 83 |
-
"linear_warmup_schedule",
|
| 84 |
-
"cosine_schedule",
|
| 85 |
-
"cosine_warmup_schedule",
|
| 86 |
-
"CosineWarmupScheduler",
|
| 87 |
]
|
|
|
|
| 39 |
"to_2tuple",
|
| 40 |
"to_3tuple",
|
| 41 |
"to_4tuple",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
"deactivate_requires_grad_and_to_eval",
|
| 43 |
"activate_requires_grad_and_to_train",
|
| 44 |
"update_momentum",
|
|
|
|
| 62 |
"Format",
|
| 63 |
"nchwd_to",
|
| 64 |
"nhwdc_to",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
]
|