|
@@ -26,12 +26,6 @@ class SeparateModelClassifier(Classifier):
|
|
|
|
|
|
self.separate_model = self.model.copy(mode="copy")
|
|
|
|
|
|
- def load_for_finetune(self, *args, **kwargs) -> None:
|
|
|
- super().load_for_finetune(*args, **kwargs)
|
|
|
-
|
|
|
- def load_for_inference(self, *args, **kwargs) -> None:
|
|
|
- super().load_for_inference(*args, **kwargs)
|
|
|
-
|
|
|
def load_model(self, weights_file: str, n_classes: int, *, finetune: bool = False) -> None:
|
|
|
for model in [self.model, self.separate_model]:
|
|
|
model_loader = self.get_model_loader(finetune=finetune, model=model)
|