CompVis / latent-diffusion

High-Resolution Image Synthesis with Latent Diffusion Models
MIT License
11.62k stars 1.51k forks source link

NameError: name 'trainer' is not defined #275

Open flyfaerss opened 1 year ago

flyfaerss commented 1 year ago

how can I solve this problem?

usama-axcelerate commented 1 year ago

Hello, did you find any solutions?

GrandpaXun242 commented 1 year ago

Please checkout your execute command and config file. This is my yaml config file.

model:
  base_learning_rate: 2.0e-06
  target: ldm.models.diffusion.ddpm.LatentDiffusion
  params:
    linear_start: 0.0015
    linear_end: 0.0195
    num_timesteps_cond: 1
    log_every_t: 200
    timesteps: 1000
    first_stage_key: image
    image_size: 64
    channels: 3
    monitor: val/loss_simple_ema
    unet_config:
      target: ldm.modules.diffusionmodules.openaimodel.UNetModel
      params:
        image_size: 64
        in_channels: 3
        out_channels: 3
        model_channels: 224
        attention_resolutions:
        # note: this isn\t actually the resolution but
        # the downsampling factor, i.e. this corresnponds to
        # attention on spatial resolution 8,16,32, as the
        # spatial reolution of the latents is 64 for f4
        - 8
        - 4
        - 2
        num_res_blocks: 2
        channel_mult:
        - 1
        - 2
        - 3
        - 4
        num_head_channels: 32
    first_stage_config:
      target: ldm.models.autoencoder.VQModelInterface
      params:
        ckpt_path: models/first_stage_models/vq-f4/model/model.ckpt
        embed_dim: 3
        n_embed: 8192
        ddconfig:
          double_z: false
          z_channels: 3
          resolution: 256
          in_channels: 3
          out_ch: 3
          ch: 128
          ch_mult:
          - 1
          - 2
          - 4
          num_res_blocks: 2
          attn_resolutions: []
          dropout: 0.0
        lossconfig:
          target: torch.nn.Identity
    cond_stage_config: __is_unconditional__
data:
  target: main.DataModuleFromConfig
  params:
    batch_size: 4
    num_workers: 4
    wrap: false
    train:
      target: ldm.data.lsun.LSUNBedroomsTrain
      params:
        size: 256
    validation:
      target: ldm.data.lsun.LSUNBedroomsValidation
      params:
        size: 256

lightning:
  callbacks:
    image_logger:
      target: main.ImageLogger
      params:
        batch_frequency: 70
        max_images: 3
        increase_log_steps: False

  trainer:
    benchmark: True
wwqy commented 1 year ago

Hello, I've checked my execute command and config file, which is the same as yours, but still can't solve this problem. Could you give any more solutions?

GrandpaXun242 commented 1 year ago

Hello, I've checked my execute command and config file, which is the same as yours, but still can't solve this problem. Could you give any more solutions?

hi,Give me your error information which is showed in terminal.

ray-lee-94 commented 1 year ago

I meet the problem since in the main.py, the try: except catch nothing. You should remove the try except first to find the error

cs-mshah commented 8 months ago

Facing the same issue for the cin-ldm-vq-f8.yaml config.

nickk124 commented 5 months ago

It might be an issue with this code needing an older pytorch_lightning version. Try installing version 1.6.1 (this may create other problems but should solve this.