dynamically pick number of inputs from train set based on length of first Layer