ThiennNguyen commited on
Commit
7370314
1 Parent(s): bbbe5e0

Upload training_config.yml with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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: 76
28
  temperature: 0.8
29
- top_k: 225
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: 225
79
  temperature: 0.8
80
- top_k: 225
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