sunshineatnoon / LinearStyleTransfer

This is the Pytorch implementation of "Learning Linear Transformations for Fast Image and Video Style Transfer" (CVPR 2019).
https://sites.google.com/view/linear-style-transfer-cvpr19/
BSD 2-Clause "Simplified" License
380 stars 87 forks source link

Question about the Gram matrix #17

Closed diaodeyi closed 2 years ago

diaodeyi commented 2 years ago

About the gram matrix in code: return G.div_(c*h*w) Why here use the div function and use the chw , shouldn't we use MSE directly to the matrix's elements?