bad FID of MDM trained with your implementation

[Cr7st]

Hi!

I used your implementation of MDM with the following command:

python3 tools/dist_train.sh configs/mdm/mdm_t2m_official.py logs/mdm 8 --no-validate

The result was pretty bad with an FID of 41.

However, I tested the checkpoint of mdm that you provided, the result was alright.

I'm wondering if the code of mdm was right or did I make a mistake.

Thank you!