asked
[AI Practice] Getting Started with Paper Implementation for AI Research Engineers with PyTorch
Loss (4) - Style Loss - Forward Implementation
Written on
·
40
0
논문에 나온대로 style representation의 normalization을 진행한다면
gram_matrix에서 div는 삭제하고 mseLoss를 구한 후 4*b*c^2*(h*w)^2를 수행해주면 되는걸까요
Answer