neurovlm.training.train_mlp_contrastive

neurovlm.training.train_mlp_contrastive#

neurovlm.training.train_mlp_contrastive(config, *, provider, model=None, text_projection=None, brain_projection=None)[source]#
Parameters:
  • config (MLPContrastiveTrainConfig)

  • provider (Any)

  • model (MLPContrastiveModel | None)

  • text_projection (Module | None)

  • brain_projection (Module | None)

Return type:

MLPTrainResult