[ | |
{ | |
"idx": 0, | |
"name": "Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})", | |
"path": "1_Pooling", | |
"type": "models.Pooling" | |
}, | |
{ | |
"idx": 1, | |
"name": "Dense({'in_features': 768, 'out_features': 768, 'bias': False, 'activation_function': 'torch.nn.modules.activation.Tanh'})", | |
"path": "2_Dense", | |
"type": "models.Dense" | |
}, | |
{ | |
"idx": 2, | |
"name": "Normalize()", | |
"path": "3_Normalize", | |
"type": "models.Normalize" | |
} | |
] |