neurovlm.training.train_mlp_brain_to_text_retrieval#
- neurovlm.training.train_mlp_brain_to_text_retrieval(config, *, provider, model=None, text_projection=None, brain_projection=None)[source]#
Train symmetric InfoNCE while selecting the brain-to-text direction.
- Parameters:
config (MLPBrainToTextRetrievalTrainConfig)
provider (Any)
model (MLPContrastiveModel | None)
text_projection (Module | None)
brain_projection (Module | None)
- Return type:
MLPTrainResult