ThiennNguyen
commited on
Commit
•
7370314
1
Parent(s):
bbbe5e0
Upload training_config.yml with huggingface_hub
Browse files- training_config.yml +4 -4
training_config.yml
CHANGED
@@ -24,9 +24,9 @@ checkpointer:
|
|
24 |
resume_from_checkpoint: false
|
25 |
interim_checkpoint_steps: 15000
|
26 |
interim_gen_steps: null
|
27 |
-
max_new_tokens:
|
28 |
temperature: 0.8
|
29 |
-
top_k:
|
30 |
dataset:
|
31 |
_component_: ds.EvenBatcher
|
32 |
buffer_size: 72
|
@@ -75,8 +75,8 @@ inference:
|
|
75 |
prompt_template: 'Video:
|
76 |
{video}
|
77 |
Caption the previous video.'
|
78 |
-
max_new_tokens:
|
79 |
temperature: 0.8
|
80 |
-
top_k:
|
81 |
quantizer: null
|
82 |
gradient-accumulation-steps: 32
|
|
|
24 |
resume_from_checkpoint: false
|
25 |
interim_checkpoint_steps: 15000
|
26 |
interim_gen_steps: null
|
27 |
+
max_new_tokens: 77
|
28 |
temperature: 0.8
|
29 |
+
top_k: 231
|
30 |
dataset:
|
31 |
_component_: ds.EvenBatcher
|
32 |
buffer_size: 72
|
|
|
75 |
prompt_template: 'Video:
|
76 |
{video}
|
77 |
Caption the previous video.'
|
78 |
+
max_new_tokens: 231
|
79 |
temperature: 0.8
|
80 |
+
top_k: 231
|
81 |
quantizer: null
|
82 |
gradient-accumulation-steps: 32
|