yuntian-deng commited on
Commit
4b0c1a8
·
1 Parent(s): c260e72

Update 2e5_debug_gpt_firstframe_posmap.yaml

Browse files
2e5_debug_gpt_firstframe_posmap.yaml CHANGED
@@ -2,7 +2,7 @@ save_path: saved_fixcursor_lr2e5_debug_gpt_firstframe_posmap
2
 
3
  model:
4
  base_learning_rate: 2.0e-05
5
- target: latent_diffusion.ldm.models.diffusion.ddpm.LatentDiffusion
6
  params:
7
  linear_start: 0.0015
8
  linear_end: 0.0195
@@ -20,7 +20,7 @@ model:
20
  monitor: val/loss_simple_ema
21
 
22
  unet_config:
23
- target: latent_diffusion.ldm.modules.diffusionmodules.openaimodel.UNetModel
24
  params:
25
  image_size: 64
26
  in_channels: 25
@@ -41,7 +41,7 @@ model:
41
  transformer_depth: 1
42
  context_dim: 768
43
  first_stage_config:
44
- target: latent_diffusion.ldm.models.autoencoder.VQModelInterface
45
  params:
46
  embed_dim: 3
47
  n_embed: 8192
@@ -65,7 +65,7 @@ model:
65
  target: torch.nn.Identity
66
 
67
  cond_stage_config:
68
- target: latent_diffusion.ldm.modules.encoders.modules.GPTEmbedder
69
  params:
70
  n_embed: 768
71
  n_layer: 12
 
2
 
3
  model:
4
  base_learning_rate: 2.0e-05
5
+ target: ldm.models.diffusion.ddpm.LatentDiffusion
6
  params:
7
  linear_start: 0.0015
8
  linear_end: 0.0195
 
20
  monitor: val/loss_simple_ema
21
 
22
  unet_config:
23
+ target: ldm.modules.diffusionmodules.openaimodel.UNetModel
24
  params:
25
  image_size: 64
26
  in_channels: 25
 
41
  transformer_depth: 1
42
  context_dim: 768
43
  first_stage_config:
44
+ target: ldm.models.autoencoder.VQModelInterface
45
  params:
46
  embed_dim: 3
47
  n_embed: 8192
 
65
  target: torch.nn.Identity
66
 
67
  cond_stage_config:
68
+ target: ldm.modules.encoders.modules.GPTEmbedder
69
  params:
70
  n_embed: 768
71
  n_layer: 12