Unverified Commit eb2f8b1f authored by ver217's avatar ver217 Committed by GitHub
Browse files

add how to build tfrecord dataset (#48)

parent 4da256a5
......@@ -47,3 +47,8 @@ Since Mixup is used as data augmentation, we define the loss function of Mixup h
`hooks.py`
We define hook functions that record running information to help debugging.
# How to build TFRecords dataset
As we use [DALI](https://github.com/NVIDIA/DALI) to read data, we use the TFRecords dataset instead of raw Imagenet dataset. If you don't have TFRecords dataset, follow [imagenet-tools](https://github.com/ver217/imagenet-tools) to build one.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment