ZiyuG commited on
Commit
f579282
·
verified ·
1 Parent(s): 4528ab7

Update dpo/config.yaml

Browse files
Files changed (1) hide show
  1. dpo/config.yaml +5 -5
dpo/config.yaml CHANGED
@@ -5,21 +5,21 @@ wandb:
5
  experiment:
6
  project: training
7
  name: show-o-dpo(only sft)
8
- output_dir: /ssd7.7tb2/zrr/tcz/Show-o/ckts/train_set_hq_prune_full_t2i_dpo_10k_1e-5
9
  save_every: 10000
10
  eval_every: 2500
11
  generate_every: 1000
12
  log_every: 1
13
  log_grad_norm_every: 20
14
- logging_dir: /ssd7.7tb2/zrr/tcz/Show-o/ckts/train_set_hq_prune_full_t2i_dpo_10k_1e-5/logs
15
  model:
16
  vq_model:
17
  type: magvitv2
18
  vq_model_name: showlab/magvitv2
19
- pretrained_model_path: /ssd7.7tb2/zrr/tcz/Show-o/ckts/magvitv2/pytorch_model.safetensors
20
  showo:
21
  load_from_showo: false
22
- pretrained_model_path: /ssd7.7tb2/zrr/tcz/Show-o/ckts/pytorch_model.safetensors
23
  w_clip_vit: false
24
  vocab_size: 58498
25
  llm_vocab_size: 50295
@@ -31,7 +31,7 @@ model:
31
  dataset:
32
  gen_type: t2i
33
  params:
34
- data_path: /ssd7.7tb2/zrr/tcz/Show-o/ov/t2i_dpo_draft.yaml
35
  validation_prompts_file: validation_prompts/showoprompts.txt
36
  shuffle_buffer_size: 1000
37
  num_workers: 32
 
5
  experiment:
6
  project: training
7
  name: show-o-dpo(only sft)
8
+ output_dir: train_set_hq_prune_full_t2i_dpo_10k_1e-5
9
  save_every: 10000
10
  eval_every: 2500
11
  generate_every: 1000
12
  log_every: 1
13
  log_grad_norm_every: 20
14
+ logging_dir: train_set_hq_prune_full_t2i_dpo_10k_1e-5/logs
15
  model:
16
  vq_model:
17
  type: magvitv2
18
  vq_model_name: showlab/magvitv2
19
+ pretrained_model_path: pytorch_model.safetensors
20
  showo:
21
  load_from_showo: false
22
+ pretrained_model_path: pytorch_model.safetensors
23
  w_clip_vit: false
24
  vocab_size: 58498
25
  llm_vocab_size: 50295
 
31
  dataset:
32
  gen_type: t2i
33
  params:
34
+ data_path: t2i_dpo_draft.yaml
35
  validation_prompts_file: validation_prompts/showoprompts.txt
36
  shuffle_buffer_size: 1000
37
  num_workers: 32