The visualization looks like there is bug in code

[vuthede]

Hi, thanks for share your great work.
Could I have 2 quetions for you?
1 > I have just try to run your code, but the visualization looks like there is something wrong in the code? .i.e I guess the bbox should be located in the face areas but the bbox keep moving around
Do you have any idea about that
https://drive.google.com/file/d/1NTx1qkEQxdS6Ezq4mkIWDsX2ODdhy_I4/view?usp=sharing
2> I run on the other type of dataset like LRS2 and choose the n_speaks=1, but it seems the model seems not well-transferable to this type of dataset when most of the image is the face and neck of 1 person? Do have any idea about that
https://drive.google.com/file/d/1vTkbop-xaNHYGPriv50CJ1aGEBzepcro/view?usp=sharing
Thanks

[afourast]

Hi, thanks for the feedback. There's a known bug with the flow aggregation for torch versions >=1.4. Could you please try setting up conda with the supplied environment.yml and running the demo from there? It should fix the issue. I'll patch the problem soon.