ydshieh commited on
Commit
212ecaa
·
1 Parent(s): aebbb72
Files changed (1) hide show
  1. run_image_captioning_flax.py +3 -1
run_image_captioning_flax.py CHANGED
@@ -116,9 +116,11 @@ class TrainingArguments:
116
  """
117
  Split a dataset into chunks of size `block_size`. On each block, images are transformed by the feature extractor
118
  and are kept in memory, and the batches of size `batch_size` are yield before processing the next block.
 
 
119
  """
120
  block_size: int = field(
121
- default=64,
122
  metadata={"help": _block_size_doc}
123
  )
124
  learning_rate: float = field(default=5e-5, metadata={"help": "The initial learning rate for AdamW."})
 
116
  """
117
  Split a dataset into chunks of size `block_size`. On each block, images are transformed by the feature extractor
118
  and are kept in memory, and the batches of size `batch_size` are yield before processing the next block.
119
+ The default value `0` indicates we don't use blocks, and the whole dataset will be preprocessed
120
+ (tokenization + feature extraction) and cached before training.
121
  """
122
  block_size: int = field(
123
+ default=0,
124
  metadata={"help": _block_size_doc}
125
  )
126
  learning_rate: float = field(default=5e-5, metadata={"help": "The initial learning rate for AdamW."})