mskcc/mimsi

Use Attention to False: dimension fix

giakoumoglou opened this issue · 2 comments

mimsi/model/mi_msi_model.py line 164 correct code: S = torch.mean(I, 0)

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?

No problem, I confused lines