From eb2f8b1f6bc99a2a09519216090129568cced5dc Mon Sep 17 00:00:00 2001 From: ver217 Date: Thu, 2 Dec 2021 16:31:23 +0800 Subject: [PATCH] add how to build tfrecord dataset (#48) --- examples/vit-b16/README.md | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/examples/vit-b16/README.md b/examples/vit-b16/README.md index 844867db6..4437ea060 100644 --- a/examples/vit-b16/README.md +++ b/examples/vit-b16/README.md @@ -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. +