| 215 | |
| 216 | |
| 217 | def dataset_train(epoch_id, dataset, fetch_vars, exe, config): |
| 218 | #logger.info("Epoch: {}, Running Dataset Begin.".format(epoch)) |
| 219 | fetch_info = [ |
| 220 | "Epoch {} Var {}".format(epoch_id, var_name) for var_name in fetch_vars |
| 221 | ] |
| 222 | fetch_vars = [var for _, var in fetch_vars.items()] |
| 223 | print_interval = config.get("runner.print_interval") |
| 224 | exe.train_from_dataset( |
| 225 | program=paddle.static.default_main_program(), |
| 226 | dataset=dataset, |
| 227 | fetch_list=fetch_vars, |
| 228 | fetch_info=fetch_info, |
| 229 | print_period=print_interval, |
| 230 | debug=config.get("runner.dataset_debug")) |
| 231 | |
| 232 | |
| 233 | def dataloader_train(epoch_id, train_dataloader, input_data_names, fetch_vars, |