hustvl
diff --git a/‎.gitignore
+3 b/‎.gitignore
+3
diff --git a/‎base_ml/__pycache__/base_experiment.cpython-310.pyc
-14.3 KB b/‎base_ml/__pycache__/base_experiment.cpython-310.pyc
-14.3 KB
diff --git a/‎base_ml/__pycache__/base_experiment.cpython-38.pyc
-14.3 KB b/‎base_ml/__pycache__/base_experiment.cpython-38.pyc
-14.3 KB
diff --git a/‎base_ml/__pycache__/base_optim.cpython-310.pyc
-504 Bytes b/‎base_ml/__pycache__/base_optim.cpython-310.pyc
-504 Bytes
diff --git a/‎base_ml/__pycache__/base_validator.cpython-310.pyc
-394 Bytes b/‎base_ml/__pycache__/base_validator.cpython-310.pyc
-394 Bytes
diff --git a/‎base_ml/__pycache__/optim_factory.cpython-310.pyc
-4.72 KB b/‎base_ml/__pycache__/optim_factory.cpython-310.pyc
-4.72 KB
diff --git a/‎cell_segmentation/__pycache__/__init__.cpython-310.pyc
-157 Bytes b/‎cell_segmentation/__pycache__/__init__.cpython-310.pyc
-157 Bytes
diff --git a/‎cell_segmentation/datasets/__pycache__/base_cell.cpython-310.pyc
-3.2 KB b/‎cell_segmentation/datasets/__pycache__/base_cell.cpython-310.pyc
-3.2 KB
diff --git a/‎cell_segmentation/datasets/__pycache__/base_cell.cpython-38.pyc
-3.22 KB b/‎cell_segmentation/datasets/__pycache__/base_cell.cpython-38.pyc
-3.22 KB
diff --git a/‎cell_segmentation/datasets/__pycache__/cell_graph_datamodel.cpython-38.pyc
-680 Bytes b/‎cell_segmentation/datasets/__pycache__/cell_graph_datamodel.cpython-38.pyc
-680 Bytes
diff --git a/‎cell_segmentation/datasets/__pycache__/conic.cpython-310.pyc
-7.55 KB b/‎cell_segmentation/datasets/__pycache__/conic.cpython-310.pyc
-7.55 KB
diff --git a/‎cell_segmentation/datasets/__pycache__/conic.cpython-38.pyc
-7.56 KB b/‎cell_segmentation/datasets/__pycache__/conic.cpython-38.pyc
-7.56 KB
diff --git a/‎cell_segmentation/datasets/__pycache__/dataset_coordinator.cpython-310.pyc
-1.81 KB b/‎cell_segmentation/datasets/__pycache__/dataset_coordinator.cpython-310.pyc
-1.81 KB
diff --git a/‎cell_segmentation/datasets/__pycache__/dataset_coordinator.cpython-38.pyc
-1.8 KB b/‎cell_segmentation/datasets/__pycache__/dataset_coordinator.cpython-38.pyc
-1.8 KB
diff --git a/‎cell_segmentation/datasets/__pycache__/monuseg.cpython-38.pyc
-4.13 KB b/‎cell_segmentation/datasets/__pycache__/monuseg.cpython-38.pyc
-4.13 KB
diff --git a/‎cell_segmentation/datasets/__pycache__/pannuke.cpython-310.pyc
-15.8 KB b/‎cell_segmentation/datasets/__pycache__/pannuke.cpython-310.pyc
-15.8 KB
diff --git a/‎cell_segmentation/datasets/__pycache__/pannuke.cpython-38.pyc
-15.8 KB b/‎cell_segmentation/datasets/__pycache__/pannuke.cpython-38.pyc
-15.8 KB
diff --git a/‎cell_segmentation/experiments/__pycache__/__init__.cpython-310.pyc
-166 Bytes b/‎cell_segmentation/experiments/__pycache__/__init__.cpython-310.pyc
-166 Bytes
diff --git a/‎cell_segmentation/experiments/__pycache__/__init__.cpython-38.pyc
-154 Bytes b/‎cell_segmentation/experiments/__pycache__/__init__.cpython-38.pyc
-154 Bytes
diff --git a/‎cell_segmentation/experiments/__pycache__/experiment_cellvit_conic.cpython-310.pyc
-22.5 KB b/‎cell_segmentation/experiments/__pycache__/experiment_cellvit_conic.cpython-310.pyc
-22.5 KB
diff --git a/‎cell_segmentation/experiments/__pycache__/experiment_cellvit_conic.cpython-38.pyc
-22.5 KB b/‎cell_segmentation/experiments/__pycache__/experiment_cellvit_conic.cpython-38.pyc
-22.5 KB
diff --git a/‎cell_segmentation/experiments/__pycache__/experiment_cellvit_pannuke.cpython-310.pyc
-22.7 KB b/‎cell_segmentation/experiments/__pycache__/experiment_cellvit_pannuke.cpython-310.pyc
-22.7 KB
diff --git a/‎cell_segmentation/experiments/__pycache__/experiment_cellvit_pannuke.cpython-38.pyc
-24.4 KB b/‎cell_segmentation/experiments/__pycache__/experiment_cellvit_pannuke.cpython-38.pyc
-24.4 KB
diff --git a/‎cell_segmentation/inference/__pycache__/__init__.cpython-310.pyc
-167 Bytes b/‎cell_segmentation/inference/__pycache__/__init__.cpython-310.pyc
-167 Bytes
diff --git a/‎cell_segmentation/inference/__pycache__/__init__.cpython-38.pyc
-152 Bytes b/‎cell_segmentation/inference/__pycache__/__init__.cpython-38.pyc
-152 Bytes
diff --git a/‎cell_segmentation/inference/__pycache__/cell_detection.cpython-38.pyc
-29.4 KB b/‎cell_segmentation/inference/__pycache__/cell_detection.cpython-38.pyc
-29.4 KB
diff --git a/‎cell_segmentation/inference/__pycache__/inference_cellvit_experiment_pannuke.cpython-310.pyc
-30.1 KB b/‎cell_segmentation/inference/__pycache__/inference_cellvit_experiment_pannuke.cpython-310.pyc
-30.1 KB
diff --git a/‎cell_segmentation/inference/__pycache__/inference_cellvit_experiment_pannuke.cpython-38.pyc
-30.4 KB b/‎cell_segmentation/inference/__pycache__/inference_cellvit_experiment_pannuke.cpython-38.pyc
-30.4 KB
diff --git a/‎cell_segmentation/trainer/__pycache__/__init__.cpython-310.pyc
-162 Bytes b/‎cell_segmentation/trainer/__pycache__/__init__.cpython-310.pyc
-162 Bytes
diff --git a/‎cell_segmentation/trainer/__pycache__/__init__.cpython-38.pyc
-150 Bytes b/‎cell_segmentation/trainer/__pycache__/__init__.cpython-38.pyc
-150 Bytes
diff --git a/‎cell_segmentation/trainer/__pycache__/trainer_cellvit.cpython-310.pyc
-23.7 KB b/‎cell_segmentation/trainer/__pycache__/trainer_cellvit.cpython-310.pyc
-23.7 KB
diff --git a/‎cell_segmentation/trainer/__pycache__/trainer_cellvit.cpython-38.pyc
-23.7 KB b/‎cell_segmentation/trainer/__pycache__/trainer_cellvit.cpython-38.pyc
-23.7 KB
diff --git a/‎cell_segmentation/utils/__pycache__/__init__.cpython-310.pyc
-163 Bytes b/‎cell_segmentation/utils/__pycache__/__init__.cpython-310.pyc
-163 Bytes
diff --git a/‎cell_segmentation/utils/__pycache__/__init__.cpython-38.pyc
-148 Bytes b/‎cell_segmentation/utils/__pycache__/__init__.cpython-38.pyc
-148 Bytes
diff --git a/‎cell_segmentation/utils/__pycache__/metrics.cpython-310.pyc
-5.58 KB b/‎cell_segmentation/utils/__pycache__/metrics.cpython-310.pyc
-5.58 KB
diff --git a/‎cell_segmentation/utils/__pycache__/metrics.cpython-38.pyc
-5.68 KB b/‎cell_segmentation/utils/__pycache__/metrics.cpython-38.pyc
-5.68 KB
diff --git a/‎cell_segmentation/utils/__pycache__/post_proc_cellvit.cpython-310.pyc
-8.66 KB b/‎cell_segmentation/utils/__pycache__/post_proc_cellvit.cpython-310.pyc
-8.66 KB
diff --git a/‎cell_segmentation/utils/__pycache__/post_proc_cellvit.cpython-38.pyc
-8.69 KB b/‎cell_segmentation/utils/__pycache__/post_proc_cellvit.cpython-38.pyc
-8.69 KB
diff --git a/‎cell_segmentation/utils/__pycache__/template_geojson.cpython-38.pyc
-921 Bytes b/‎cell_segmentation/utils/__pycache__/template_geojson.cpython-38.pyc
-921 Bytes
diff --git a/‎cell_segmentation/utils/__pycache__/tools.cpython-310.pyc
-10.3 KB b/‎cell_segmentation/utils/__pycache__/tools.cpython-310.pyc
-10.3 KB
diff --git a/‎cell_segmentation/utils/__pycache__/tools.cpython-38.pyc
-10.3 KB b/‎cell_segmentation/utils/__pycache__/tools.cpython-38.pyc
-10.3 KB
diff --git a/‎models/__pycache__/__init__.cpython-310.pyc
-146 Bytes b/‎models/__pycache__/__init__.cpython-310.pyc
-146 Bytes
diff --git a/‎models/segmentation/__pycache__/__init__.cpython-310.pyc
-159 Bytes b/‎models/segmentation/__pycache__/__init__.cpython-310.pyc
-159 Bytes
diff --git a/‎models/segmentation/cell_segmentation/__pycache__/__init__.cpython-310.pyc
-2 Bytes b/‎models/segmentation/cell_segmentation/__pycache__/__init__.cpython-310.pyc
-2 Bytes
diff --git a/‎models/segmentation/cell_segmentation/__pycache__/cellvit.cpython-310.pyc
60 Bytes b/‎models/segmentation/cell_segmentation/__pycache__/cellvit.cpython-310.pyc
60 Bytes
diff --git a/‎models/segmentation/cell_segmentation/__pycache__/cellvit_unirepLKnet.cpython-310.pyc
-2 Bytes b/‎models/segmentation/cell_segmentation/__pycache__/cellvit_unirepLKnet.cpython-310.pyc
-2 Bytes
diff --git a/‎models/segmentation/cell_segmentation/cellvit.py
+2-1 b/‎models/segmentation/cell_segmentation/cellvit.py
+2-1
diff --git a/‎utils/__pycache__/__init__.cpython-310.pyc
-246 Bytes b/‎utils/__pycache__/__init__.cpython-310.pyc
-246 Bytes
diff --git a/‎utils/__pycache__/logger.cpython-310.pyc
-3.66 KB b/‎utils/__pycache__/logger.cpython-310.pyc
-3.66 KB
diff --git a/‎utils/__pycache__/tools.cpython-310.pyc
-7.38 KB b/‎utils/__pycache__/tools.cpython-310.pyc
-7.38 KB
@@ -0,0 +1,3 @@
+**/__pycache__
+/.vscode
+test.py
@@ -21,6 +21,7 @@
 from .cellvit_unirepLKnet import UniRepLKNet
 from .replknet import  *
 
+from huggingface_hub import PyTorchModelHubMixin
 
 
 class LayerNorm(nn.Module):
@@ -258,7 +259,7 @@ def __init__(
 
 
 
-class CellViT(UniRepLKNet):
+class CellViT(UniRepLKNet,PyTorchModelHubMixin):
 
     def __init__(
         self,
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+**/__pycache__`
	`2`	`+/.vscode`
	`3`	`+test.py`