where is the code for local attention loss?

Question

where is the code for local attention loss?

ZhendongWang6 opened this issue a year ago · 1 comments

Hi, it is a great work! But I do not find the implementation of the local attention loss. Could you help me? Thanks.

Answer 1 · 2023-12-20T13:13:58.000Z

Much appreciated. The implementation of the local attention loss can be found in ./sgm/modules/diffusionmodules/loss.py. Check the function get_local_loss().