qdrant/quaterion

Implement a `nn.Sequential`-like head layer

monatis opened this issue · 1 comments

monatis commented

This will allow users to compose custom head layers easily by appending arbitrary nn.Modules to it.

For example, they will be able to add a dropout before the actual head layer etc.