askerlee/segtran

Where can we find the code that generates the prototype?

saklanip opened this issue · 1 comments

saklanip commented
Where can we find the code that generates the prototype?
askerlee commented

Good question. I should have explained it in the readme page. They are actually attactors in the squeeze-and-expansion transformer layer:

segtran/code/networks/segtran_shared.py

Line 499 in 0515f5c

self.attractors = Parameter(torch.randn(1, self.num_attractors, self.in_feat_dim))