Lyosha

It has something to do with the way the LDMs are written, I removed the `trainer.test` call from `main.py` and added a `--no_test`, but it didn't do it, I still...

Any progress on this? Would be curious if you managed to fix this @xiaofei12306 @zjz894251se ? Thanks!