georgian-io/Multimodal-Toolkit

There is a small mistake in tabular_combiner.py

weibobo2015 opened this issue · 0 comments

As the title says, there is a mistake in tabular_combiner.py.
Line 409:
g_num = (torch.cat([w_text, w_cat], dim=-1) * self.weight_a).sum(dim=1).unsqueeze(0).T

should change to:
g_num = (torch.cat([w_text, w_num], dim=-1) * self.weight_a).sum(dim=1).unsqueeze(0).T

because it is in a numerical part which should use w_num instead of w_cat