langml.baselines.contrastive.cli

Module Contents

Functions

contrastive()

contrastive learning command line tools

simcse(backbone: str, epoch: int, batch_size: int, learning_rate: float, dropout_rate: float, temperature: float, pooling_strategy: str, max_len: Optional[int], early_stop: int, monitor: str, lowercase: bool, tokenizer_type: Optional[str], config_path: str, ckpt_path: str, vocab_path: str, train_path: str, test_path: str, save_dir: str, verbose: int, apply_aeda: bool, aeda_language: str, do_evaluate: bool, distributed_training: bool, distributed_strategy: str)

langml.baselines.contrastive.cli.contrastive()[source]

contrastive learning command line tools

langml.baselines.contrastive.cli.simcse(backbone: str, epoch: int, batch_size: int, learning_rate: float, dropout_rate: float, temperature: float, pooling_strategy: str, max_len: Optional[int], early_stop: int, monitor: str, lowercase: bool, tokenizer_type: Optional[str], config_path: str, ckpt_path: str, vocab_path: str, train_path: str, test_path: str, save_dir: str, verbose: int, apply_aeda: bool, aeda_language: str, do_evaluate: bool, distributed_training: bool, distributed_strategy: str)[source]