|
@@ -269,7 +269,7 @@ class Main(object):
|
|
|
self.count_method = self.config.get("runner.example_count_method",
|
|
|
"example")
|
|
|
|
|
|
- def dataset_train_loop(self, epoch):
|
|
|
+ def dataset_train_loop(self, epoch=0):
|
|
|
logger.info("Epoch: {}, Running Dataset Begin.".format(epoch))
|
|
|
|
|
|
fetch_info = [
|