(model, mode)
| 468 | |
| 469 | |
| 470 | def _get_model_feed(model, mode): |
| 471 | if mode == ModeKeys.PREDICT: |
| 472 | feed = model._feed_inputs |
| 473 | else: |
| 474 | feed = ( |
| 475 | model._feed_inputs + model._feed_targets + model._feed_sample_weights) |
| 476 | return feed |
| 477 | |
| 478 | |
| 479 | def _print_train_info(num_samples_or_steps, val_samples_or_steps, is_dataset): |