Update 2e5_debug_gpt_firstframe_posmap_debugidentity_256.yaml

2e5_debug_gpt_firstframe_posmap_debugidentity_256.yaml

@@ -2,7 +2,7 @@ save_path: saved_fixcursor_lr2e5_debug_gpt_firstframe_posmap_debugidentity_256
 
 model:
   base_learning_rate: 2.0e-05
-  target: latent_diffusion.ldm.models.diffusion.ddpm.LatentDiffusion
+  target: ldm.models.diffusion.ddpm.LatentDiffusion
   params:
     linear_start: 0.0015
     linear_end: 0.0195
@@ -11,7 +11,6 @@ model:
     timesteps: 1000
     first_stage_key: image
     cond_stage_key: action_
-    scheduler_sampling_rate: 0.0
     hybrid_key: c_concat
     image_size: 256
     channels: 3
@@ -20,7 +19,7 @@ model:
     monitor: val/loss_simple_ema
 
     unet_config:
-      target: latent_diffusion.ldm.modules.diffusionmodules.openaimodel.UNetModel
+      target: ldm.modules.diffusionmodules.openaimodel.UNetModel
       params:
         image_size: 256
         in_channels: 25
@@ -41,13 +40,13 @@ model:
         transformer_depth: 1
         context_dim: 768
     first_stage_config:
-      target: latent_diffusion.ldm.models.autoencoder.IdentityFirstStage
+      target: ldm.models.autoencoder.IdentityFirstStage
       params:
         embed_dim: 3
         n_embed: 8192
 
     cond_stage_config:
-      target: latent_diffusion.ldm.modules.encoders.modules.GPTEmbedder
+      target: ldm.modules.encoders.modules.GPTEmbedder
       params:
         n_embed: 768
         n_layer: 12