zhyhan / AD3DMIL

Attention Based Deep 3D Multiple Instance Learning
MIT License
23 stars 6 forks source link

Version with Gated Attention in ENModel #4

Closed JBotzev closed 3 months ago

JBotzev commented 3 months ago

I noticed that the provided implementation uses mean pooling. I am particularly interested in the version of the model where the gated attention mechanism is properly integrated. Is there an existing version or could you provide some guidance regarding its implementation ? Any help would be greatly appreciated.