delete start_epocj

paddlex/cv/models/base.py

@@ -364,7 +364,6 @@ class BaseAPI:
                    num_epochs,
                    train_dataset,
                    train_batch_size,
-                   start_epoch=0,
                    eval_dataset=None,
                    save_interval_epochs=1,
                    log_interval_steps=10,
@@ -439,6 +438,7 @@ class BaseAPI:
         best_accuracy_key = ""
         best_accuracy = -1.0
         best_model_epoch = -1
+        start_epoch = self.completed_epochs
         for i in range(start_epoch, num_epochs):
             records = list()
             step_start_time = time.time()

paddlex/cv/models/classifier.py

@@ -164,12 +164,8 @@ class BaseClassifier(BaseAPI):
             sensitivities_file=sensitivities_file,
             eval_metric_loss=eval_metric_loss,
             resume_checkpoint=resume_checkpoint)
-        start_epoch = 0
-        if resume_checkpoint:
-            start_epoch = self.completed_epochs
         # 训练
         self.train_loop(
-            start_epoch=start_epoch,
             num_epochs=num_epochs,
             train_dataset=train_dataset,
             train_batch_size=train_batch_size,

paddlex/cv/models/deeplabv3p.py

@@ -288,12 +288,8 @@ class DeepLabv3p(BaseAPI):
             sensitivities_file=sensitivities_file,
             eval_metric_loss=eval_metric_loss,
             resume_checkpoint=resume_checkpoint)
-        start_epoch = 0
-        if resume_checkpoint:
-            start_epoch = self.completed_epochs
         # 训练
         self.train_loop(
-            start_epoch=start_epoch,
             num_epochs=num_epochs,
             train_dataset=train_dataset,
             train_batch_size=train_batch_size,

paddlex/cv/models/faster_rcnn.py

@@ -235,12 +235,9 @@ class FasterRCNN(BaseAPI):
             fuse_bn=fuse_bn,
             save_dir=save_dir,
             resume_checkpoint=resume_checkpoint)
-        start_epoch = 0
-        if resume_checkpoint:
-            start_epoch = self.completed_epochs
+
         # 训练
         self.train_loop(
-            start_epoch=start_epoch,
             num_epochs=num_epochs,
             train_dataset=train_dataset,
             train_batch_size=train_batch_size,

paddlex/cv/models/mask_rcnn.py

@@ -202,12 +202,8 @@ class MaskRCNN(FasterRCNN):
             fuse_bn=fuse_bn,
             save_dir=save_dir,
             resume_checkpoint=resume_checkpoint)
-        start_epoch = 0
-        if resume_checkpoint:
-            start_epoch = self.completed_epochs
         # 训练
         self.train_loop(
-            start_epoch=start_epoch,
             num_epochs=num_epochs,
             train_dataset=train_dataset,
             train_batch_size=train_batch_size,

paddlex/cv/models/yolo_v3.py

@@ -240,12 +240,8 @@ class YOLOv3(BaseAPI):
             sensitivities_file=sensitivities_file,
             eval_metric_loss=eval_metric_loss,
             resume_checkpoint=resume_checkpoint)
-        start_epoch = 0
-        if resume_checkpoint:
-            start_epoch = self.completed_epochs
         # 训练
         self.train_loop(
-            start_epoch=start_epoch,
             num_epochs=num_epochs,
             train_dataset=train_dataset,
             train_batch_size=train_batch_size,