From 9824ac53a60f3226b96fee1539ae99005b1fe0f7 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Sat, 18 Sep 2021 09:42:24 +0000
Subject: [PATCH] fix bugs

---
 deepspeech/training/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepspeech/training/trainer.py b/deepspeech/training/trainer.py
index 22dded6fa..ffc52775a 100644
--- a/deepspeech/training/trainer.py
+++ b/deepspeech/training/trainer.py
@@ -200,7 +200,7 @@ class Trainer():
             if isinstance(batch_sampler, paddle.io.DistributedBatchSampler):
                 batch_sampler.set_epoch(self.epoch)
 
-    def before_train(self, from_scratch):
+    def before_train(self):
         from_scratch = self.resume_or_scratch()
         if from_scratch:
             # scratch: save init model, i.e. 0 epoch