nikkozzblu

I had a similar issue and came to the conclusion there was a typo in the code: https://github.com/johannakarras/DreamPose/blob/5bf30b7df70cf6f2e0bb25556c6ff2cbf0f2b1bf/finetune-unet.py#L191 should be `frame_j = [example["frame_j"] for example in examples]` Working much better...