zizhaozhang

import argparse
import psutil
import tensorflow as tf
from typing import Dict, Any, Callable, Tuple

## Data Input Function
def data_input_fn(data_param,
                  batch_size:int=None,
                  shuffle=False) -> Callable[[], Tuple]:
    """Return the input function to get the test data.