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:
|
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:
|
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:
|