search

nihaomiao / CVPR23_LFDM

The pytorch implementation of our CVPR 2023 paper "Conditional Image-to-Video Generation with Latent Flow Diffusion Models"
BSD 2-Clause "Simplified" License
443 stars 42 forks source link

KeyError: 'bottle_neck_feat' #4

Closed Yueming6568 closed 1 year ago

Yueming6568 commented 1 year ago

Thank you for your excellent work ! I have a problem below.

When I am trying to train DM model and running "python -u DM/train_video_flow_diffusion_mhad.py"， there is an error below: File "CVPR23_LFDM/DM/modules/video_flow_diffusion_model.py", line 144, in forward self.ref_img_fea = generated["bottle_neck_feat"].clone().detach() KeyError: 'bottle_neck_feat'

I cannot find the key ''bottle_neck_feat'' in the generator.py, please help me to solve it ~ Thank you very much!

nihaomiao commented 1 year ago

Hi, @Yueming6568, thanks for your comments! I have fixed the bug. Please feel free to let me know if you find other problems.

Yueming6568 commented 1 year ago

Sure : ) Thank you for your kind reply ! 👍