|
|
@@ -255,11 +255,14 @@ class Main(object):
|
|
|
|
|
|
def dataset_train_loop(self, epoch):
|
|
|
logger.info("Epoch: {}, Running Dataset Begin.".format(epoch))
|
|
|
+
|
|
|
fetch_info = [
|
|
|
"Epoch {} Var {}".format(epoch, var_name)
|
|
|
for var_name in self.metrics
|
|
|
]
|
|
|
+ logger.info("Metrics keys: {}".format(self.metrics.keys()))
|
|
|
fetch_vars = [var for _, var in self.metrics.items()]
|
|
|
+ logger.info("Fetch vars: {}".format(fetch_vars))
|
|
|
print_step = int(config.get("runner.print_interval"))
|
|
|
|
|
|
debug = config.get("runner.dataset_debug", False)
|
|
|
@@ -272,7 +275,7 @@ class Main(object):
|
|
|
"dump_fields": config.get("runner.dump_fields")
|
|
|
})
|
|
|
logger.info(paddle.static.default_main_program()._fleet_opt)
|
|
|
- logger.info("Epoch: {}, debug debug debug debug.".format(epoch))
|
|
|
+
|
|
|
self.exe.train_from_dataset(
|
|
|
program=paddle.static.default_main_program(),
|
|
|
dataset=self.reader,
|
|
|
@@ -280,7 +283,7 @@ class Main(object):
|
|
|
fetch_info=fetch_info,
|
|
|
print_period=print_step,
|
|
|
debug=debug)
|
|
|
- logger.info("Epoch: {}, debug1 debug1 debug1 debug1.".format(epoch))
|
|
|
+
|
|
|
|
|
|
def heter_train_loop(self, epoch):
|
|
|
logger.info(
|
often