• Asim Shankar's avatar
    [mnist]: Use FixedLengthRecordDatatest · 73def645
    Asim Shankar authored
    - Prior to this change, the use of tf.data.Dataset essentially embedded
      the entire training/evaluation dataset into the graph as a constant,
      leading to unnecessarily humungous graphs (Fixes #3017)
    - Also, use batching on the evaluation dataset to allow
      evaluation on GPUs that cannot fit the entire evaluation dataset in
      memory (Fixes #3046)
    73def645
dataset.py 3.92 KB