search

CyberAgentAILab / layout-dm

LayoutDM: Discrete Diffusion Model for Controllable Layout Generation [Inoue+, CVPR2023]
https://cyberagentailab.github.io/layout-dm
Apache License 2.0
211 stars 23 forks source link

About training log #20

Closed lifengheng closed 1 year ago

lifengheng commented 1 year ago

Hi, thanks for your great work. I follow the instructions in custom_dataset.md to train layoutDM on custom dataset, but I find the loss is high when the training is over. The loss descends from 15 to 12 after 20 epoches and fluctuates between 13 and 12 in the next 30 epoches. Is this normal, Could you share the training log ? Thanks a lot.

naoto0804 commented 1 year ago

Hi, thank you for your interest in our work! Sorry, I didn't have the exact log, but it decreases steadily at the beginning and converges. I'm not sure whether the values are typical; Maybe the better way is to compute FID regularly during training.

lifengheng commented 1 year ago

Thanks for your quick reply. I will check FID during training.