search

eladrich / pixel2style2pixel

Official Implementation for "Encoding in Style: a StyleGAN Encoder for Image-to-Image Translation" (CVPR 2021) presenting the pixel2style2pixel (pSp) framework
https://eladrich.github.io/pixel2style2pixel/
MIT License
3.19k stars 570 forks source link

the key ['loss'] in the loss_dict #127

Closed diaodeyi closed 3 years ago

diaodeyi commented 3 years ago

In line 108 and 109 of coach ,i think there is some key error in the " val_loss_dict['loss']"

yuval-alaluf commented 3 years ago

Hmm. I'm a bit surprised that you got an error there. Any chance you can provide more details? I'm assuming you're working with an older version of the code, but are you referring to these lines? https://github.com/eladrich/pixel2style2pixel/blob/72f0c04a890adb9c8a406f879980572ac600cf29/training/coach.py#L100-L102

diaodeyi commented 3 years ago

Hmm. I'm a bit surprised that you got an error there. Any chance you can provide more details? I'm assuming you're working with an older version of the code, but are you referring to these lines? https://github.com/eladrich/pixel2style2pixel/blob/72f0c04a890adb9c8a406f879980572ac600cf29/training/coach.py#L100-L102

yes， i referring to these line. And i add the loss key in the dictionary , the probelm figured out