Update training_args.txt

training_args.txt

@@ -2,4 +2,4 @@ step = 16500
 _metadata = ContainerMetadata(ref_type=typing.Any, object_type=<class 'dict'>, optional=True, key=None, flags={}, flags_root=False, resolver_cache=defaultdict(<class 'dict'>, {}), key_type=typing.Any, element_type=typing.Any)
 _parent = None
 _flags_cache = {'struct': None}
-_content = {'project_name': 'coq-theorem-embedding', 'experiment_name': 'infonce-file-final-1', 'log_level': 'INFO', 'dataset': {'dataset_path': './data/', 'rankin_ds_path_statements': './data/imm/basic/Events_statements.json', 'rankin_ds_path_references': './validationSet/reference_premises.json', 'samples_from_single_anchor': 150, 'train_split': 0.8, 'val_split': 0.2, 'test_split': 0.0}, 'base_model_name': 'microsoft/codebert-base', 'max_seq_length': 128, 'embedding_dim': 768, 'threshold_pos': 0.3, 'threshold_neg': 0.65, 'threshold_hard_neg': 0.45, 'k_negatives': 4, 'steps': 30000, 'batch_size': 32, 'warmup_ratio': 0.1, 'learning_rate': 4e-06, 'random_seed': 52, 'wandb': {'enabled': True, 'project': 'coq-embeddings', 'entity': 'kozyrev-andreiii2016', 'tags': ['dyn-lr', 'InfoNCE', 'hard-negatives', 'codebert']}, 'evaluation': {'output_dir': './checkpoints/', 'save_freq': 300, 'eval_steps': 200, 'evaluate_freq': 100, 'k_values': [5, 10], 'f_score_beta': 1, 'query_size_in_eval': 20}}
+_content = {'project_name': 'coq-theorem-embedding', 'experiment_name': 'infonce-file-final-1', 'log_level': 'INFO', 'dataset': {'dataset_path': './data/', 'rankin_ds_path_statements': './data/imm/basic/Events_statements.json', 'rankin_ds_path_references': './validationSet/reference_premises.json', 'samples_from_single_anchor': 150, 'train_split': 0.8, 'val_split': 0.2, 'test_split': 0.0}, 'base_model_name': 'microsoft/codebert-base', 'max_seq_length': 128, 'embedding_dim': 768, 'threshold_pos': 0.3, 'threshold_neg': 0.65, 'threshold_hard_neg': 0.45, 'k_negatives': 4, 'steps': 30000, 'batch_size': 32, 'warmup_ratio': 0.1, 'learning_rate': 4e-06, 'random_seed': 52, 'wandb': {'enabled': True, 'project': 'coq-embeddings', 'tags': ['dyn-lr', 'InfoNCE', 'hard-negatives', 'codebert']}, 'evaluation': {'output_dir': './checkpoints/', 'save_freq': 300, 'eval_steps': 200, 'evaluate_freq': 100, 'k_values': [5, 10], 'f_score_beta': 1, 'query_size_in_eval': 20}}