AliaksandrSiarohin / first-order-model

This repository contains the source code for the paper First Order Motion Model for Image Animation
https://aliaksandrsiarohin.github.io/first-order-model-website/
MIT License
14.55k stars 3.22k forks source link

Why use synchronized BatchNorm? #350

Open alievk opened 3 years ago

alievk commented 3 years ago

Hi Aliaksandr,

What's the point to use SynchronizedBatchNorm2d instead of vanilla BatchNorm2d?

AliaksandrSiarohin commented 3 years ago

In my experience it usually works better, so I start with sync bn. I did not test it with the latest model though, so not sure if it better in this case.

alievk commented 3 years ago

You mean it generally works better for training deep learning models or particularly for FOMM?

AliaksandrSiarohin commented 3 years ago

In general.

alievk commented 3 years ago

I only see a difference in equivariance_jacobian (pink w/ SyncBatchNorm, green w/o)

image

However, SyncBatchNorm is slow and memory consuming.

alievk commented 3 years ago

Did you explore the effect of the learning rate on the KP detector? Is the value defined in the config optimal?

AliaksandrSiarohin commented 3 years ago

Not really.