ydshieh
commited on
Commit
·
212ecaa
1
Parent(s):
aebbb72
fix
Browse files
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=
|
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."})
|