flags.py 文件源码

def register_dataset_flags():
        logging.info("Registering Dataset flags")
        flags.DEFINE_integer("batch_size", 128,
                             "Size of the batch of the dataset iterator.")

        flags.DEFINE_integer("buffer_size", 512,
                             "Size of the buffer of the dataset iterator.")

        flags.DEFINE_integer("take_count", -1,
                             "Creates a `Dataset` with at most `count` batches from this dataset.")

        flags.DEFINE_string("train_subdir", "train",
                            "Location of training TFRecords, with the training set dir.")

        flags.DEFINE_string("eval_subdir", "eval",
                            "Location of eval TFRecords, with the training set dir.")