1.
loader.py
flags.DEFINE_integer('min_stride', 3, 'Minimum stride for sequence.')
flags.DEFINE_integer('max_stride', 5, 'Maximum stride for sequence.')
该参数会影响程序能否运行,具体体现在loader.py中:
def prepare_for_training(sequences):
"""Steps applied to training dataset only."""
# Random shuffling, random subsequences and random reversal for training.
# Also we make it repeat indefinitely.
shuffled = sequences.shuffle(1000).repeat(epochs)
# Discard sequences that are too short to generate a subsequence at
# max stride.
required_length = (sequence_length - 1) * max_stride + 1
filtered = shuffled.filter(
lambda sequence: tf.greater_equal(sequence.length(), required_length))
subsequences = filtered.map(
lambda sequence: sequence.random_subsequence(sequence_length, min_stride, max_stride)
)