Use Attention to False: dimension fix
giakoumoglou opened this issue · 2 comments
giakoumoglou commented
mimsi/model/mi_msi_model.py line 164 correct code: S = torch.mean(I, 0)
john-ziegler commented
Hi Nikos - what's the issue that you're seeing here? Line 164 is A = nn.functional.softmax(A, dim=1)
, and the mean function that you have shown is already present here at line 169. Perhaps there is a versioning issue?
giakoumoglou commented
No problem, I confused lines