@@ -34,7 +34,7 @@ class DefaultFinetuner(mixins._ModelMixin, mixins._DatasetMixin, mixins._Trainer
self.init_datasets(opts)
self.init_iterators(opts)
- self.wrap_model(opts)
+ self.init_classifier(opts)
self.load_model_weights(opts)
self.init_optimizer(opts)
@@ -32,7 +32,7 @@ class _ModelMixin(abc.ABC):
def model_info(self):
return self.data_info.MODELS[self.model_type]
- def wrap_model(self, opts):
+ def init_classifier(self, opts):
clf_class, kwargs = self.classifier_cls, self.classifier_kwargs