Deep learning for NeuroImaging in Python.

Note

This page is a reference documentation. It only explains the class signature, and not how to use it. Please refer to the gallery for the big picture.

class nidl.estimators.ssl.utils.projection_heads.YAwareProjectionHead(input_dim: int = 2048, hidden_dim: int = 512, output_dim: int = 128)[source]¶

Bases: ProjectionHead

Projection head used for yAware.

“[…] zθ2 is a vanilla multilayer perceptron as in [SimCLR, 2020]” [1]

[1] yAware, 2021, https://arxiv.org/abs/2106.08808

Initialize internal Module state, shared by both nn.Module and ScriptModule.