White Papers
13 CheXNet – Inference with Nvidia T4 on Dell EMC PowerEdge R7425
Figure 6: Overview of the Estimator Interface [5]
See the Table 4 with the Estimator’s methods and modes to call train, evaluate, or predict. The
Estimator framework invokes the model function with the mode parameter set as follows:
Table 4. Implement training, evaluation, and prediction. Source [4]
Create the Estimator:
Chexnet_classifier = tf.estimator.Estimator(
model_fn=model_function, model_dir=FLAGS.model_dir, config=run_config,
params={
'densenet_depth': FLAGS.densenet_depth,
'data_format': FLAGS.data_format,
'batch_size': FLAGS.batch_size,
'lr': lr})
Define the model function for training using transfer Learning: