Allow setting granularity for train data