Understanding the Assembly Loss in the loss function

[vifi2021]

Dear Author,

Thank you so much for the code. I am trying to understand the following part of the loss function ("assembly loss" in the paper)

SST/layer/sst_loss.py

Line 56 in b42609e

loss = -(target_pre * torch.log(input_all)).sum() / target_num_pre

In the paper (https://arxiv.org/pdf/1810.11780.pdf) equation (4). "L3" (target_union) is used in the assembly loss. But in the code implementation, "target_pre" is used.

Shouldn't we replace "target_pre" to "target_union" and "target_num_pre" to "target_num_union"?

Looking forward to your advice on this. Thank you.