justinpinkney / stable-diffusion

MIT License
1.45k stars 266 forks source link

what is the expected epoch loss #58

Open psy2013GitHub opened 1 year ago

psy2013GitHub commented 1 year ago

i try to reproduce the result, but failed here is my config: gpu(v100): 2 batch size:2 accumulate_batches: 2

after training 500 epochs, the train epoch loss is about 0.01, but the generated image is bad.

can u guys paste your training metrics, such as epoch loss etc i think this should help others a lot. thanks.

hdparmar commented 1 year ago

I'm kinda lost on the metrics as well. Do you know what all this stands for? I can guess the common ones like train/loss_epoch and stuff but what is?

train/loss_simple_epoch train/loss_vlb_epoch val/loss_simple val/loss_vlb val/loss_vlb_ema val/loss_simple_ema val/loss_ema?