Loss weight different values different from the paper

diyiiyiii / StyTR-2

StyTr2 : Image Style Transfer with Transformers

364 stars 66 forks source link

Loss weight different values different from the paper #33

Open FilippoBotti opened 5 months ago

FilippoBotti commented 5 months ago

Hi, In the paper you wrote that lamba_content is 10 and lamba_style is 7, but in the code the values are inverted. Also, lamba_l1 is 50 but in the provided code is set to 70. So what are the corrected values in order to obtain the results of the paper?