Browse Source

renamed a method

Dimitri Korsch 4 năm trước cách đây
mục cha
commit
844a158386
2 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 1 1
      cvfinetune/finetuner/base.py
  2. 1 1
      cvfinetune/finetuner/mixins/model.py

+ 1 - 1
cvfinetune/finetuner/base.py

@@ -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)

+ 1 - 1
cvfinetune/finetuner/mixins/model.py

@@ -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