shanetx commited on
Commit
111d486
1 Parent(s): 5300879

add ckpt_path in model.param

Browse files
configs/2020-11-20T12-54-32-project.yaml CHANGED
@@ -3,6 +3,7 @@ model:
3
  target: taming.models.cond_transformer.Net2NetTransformer
4
  params:
5
  cond_stage_key: depth
 
6
  transformer_config:
7
  target: taming.modules.transformer.mingpt.GPT
8
  params:
@@ -14,7 +15,7 @@ model:
14
  first_stage_config:
15
  target: taming.models.vqgan.VQModel
16
  params:
17
- ckpt_path: logs/2020-09-23T17-56-33_imagenet_vqgan/checkpoints/last.ckpt
18
  embed_dim: 256
19
  n_embed: 1024
20
  ddconfig:
@@ -39,7 +40,7 @@ model:
39
  cond_stage_config:
40
  target: taming.models.vqgan.VQModel
41
  params:
42
- ckpt_path: logs/2020-11-03T15-34-24_imagenetdepth_vqgan/checkpoints/last.ckpt
43
  embed_dim: 256
44
  n_embed: 1024
45
  ddconfig:
 
3
  target: taming.models.cond_transformer.Net2NetTransformer
4
  params:
5
  cond_stage_key: depth
6
+ ckpt_path: checkpoints/last.ckpt
7
  transformer_config:
8
  target: taming.modules.transformer.mingpt.GPT
9
  params:
 
15
  first_stage_config:
16
  target: taming.models.vqgan.VQModel
17
  params:
18
+ ckpt_path: checkpoints/last.ckpt
19
  embed_dim: 256
20
  n_embed: 1024
21
  ddconfig:
 
40
  cond_stage_config:
41
  target: taming.models.vqgan.VQModel
42
  params:
43
+ ckpt_path: checkpoints/last.ckpt
44
  embed_dim: 256
45
  n_embed: 1024
46
  ddconfig: