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

Update dpo_iter/config.yaml

Browse files
Files changed (1) hide show
  1. dpo_iter/config.yaml +5 -5
dpo_iter/config.yaml CHANGED
@@ -5,21 +5,21 @@ wandb:
5
  experiment:
6
  project: training
7
  name: show-o-dpo
8
- output_dir: /ssd7.7tb2/zrr/tcz/Show-o/ckts/train_set_hq_prune_full_t2i_dpo_10k_1e-5_v2_new_2
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_v2_new_2/logs
15
  model:
16
- rw_model: /ssd7.7tb2/zrr/tcz/Show-o/geneval/outputs/train_set_hq_prune_full
17
  vq_model:
18
  type: magvitv2
19
  vq_model_name: showlab/magvitv2
20
  showo:
21
  load_from_showo: false
22
- pretrained_model_path: /ssd7.7tb2/zrr/tcz/Show-o/ckts/train_set_hq_prune_full_t2i_dpo_10k_1e-5
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
8
+ output_dir: train_set_hq_prune_full_t2i_dpo_10k_1e-5_v2_new_2
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_v2_new_2/logs
15
  model:
16
+ rw_model: train_set_hq_prune_full
17
  vq_model:
18
  type: magvitv2
19
  vq_model_name: showlab/magvitv2
20
  showo:
21
  load_from_showo: false
22
+ pretrained_model_path: train_set_hq_prune_full_t2i_dpo_10k_1e-5
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