tangjiapeng / DiffuScene

[CVPR 2024] DiffuScene: Denoising Diffusion Models for Generative Indoor Scene Synthesis
Other
225 stars 21 forks source link

How to solve "the data has no 'description' attribute" #10

Closed franklhj closed 8 months ago

franklhj commented 1 year ago

When I try to train the text-conditioned model, it show that the data has no 'description' attribute:

File "/ mnt/16T/lhj/files/remote/for_scene_gen/DiffuScene/scene_diffusion/networks/diffusion_scene_layout_ddpm.py", line 226, in get_loss. Tokenized = self.tokenizer (sample_params ["description"], return_tensors='pt',padding=True) .to (device). KeyError: 'description'

How to solve this problem?