yuntian-deng commited on
Commit
c0aa812
·
1 Parent(s): 1948e30

Update 2e5_debug_gpt_firstframe_posmap_debugidentity.yaml

Browse files
2e5_debug_gpt_firstframe_posmap_debugidentity.yaml CHANGED
@@ -2,7 +2,7 @@ save_path: saved_fixcursor_lr2e5_debug_gpt_firstframe_posmap_debugidentity
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,13 +41,13 @@ model:
41
  transformer_depth: 1
42
  context_dim: 768
43
  first_stage_config:
44
- target: latent_diffusion.ldm.models.autoencoder.IdentityFirstStage
45
  params:
46
  embed_dim: 3
47
  n_embed: 8192
48
 
49
  cond_stage_config:
50
- target: latent_diffusion.ldm.modules.encoders.modules.GPTEmbedder
51
  params:
52
  n_embed: 768
53
  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.IdentityFirstStage
45
  params:
46
  embed_dim: 3
47
  n_embed: 8192
48
 
49
  cond_stage_config:
50
+ target: ldm.modules.encoders.modules.GPTEmbedder
51
  params:
52
  n_embed: 768
53
  n_layer: 12