02/17/2024 13:45:08 - INFO - __main__ -   device: cuda, n_gpu: 1
02/17/2024 13:45:12 - INFO - __main__ -   +------------------------------------------------------------+--------------+----------+
| Layer Name                                                 | Output Shape |  Param # |
+------------------------------------------------------------+--------------+----------+
| encoder.embeddings.word_embeddings.weight                  | [51451, 768] | 39514368 |
| encoder.embeddings.position_embeddings.weight              |  [1026, 768] |   787968 |
| encoder.embeddings.token_type_embeddings.weight            |    [10, 768] |     7680 |
| encoder.embeddings.LayerNorm.weight                        |        [768] |      768 |
| encoder.embeddings.LayerNorm.bias                          |        [768] |      768 |
| encoder.encoder.layer.0.attention.self.query.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.0.attention.self.query.bias          |        [768] |      768 |
| encoder.encoder.layer.0.attention.self.key.weight          |   [768, 768] |   589824 |
| encoder.encoder.layer.0.attention.self.key.bias            |        [768] |      768 |
| encoder.encoder.layer.0.attention.self.value.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.0.attention.self.value.bias          |        [768] |      768 |
| encoder.encoder.layer.0.attention.output.dense.weight      |   [768, 768] |   589824 |
| encoder.encoder.layer.0.attention.output.dense.bias        |        [768] |      768 |
| encoder.encoder.layer.0.attention.output.LayerNorm.weight  |        [768] |      768 |
| encoder.encoder.layer.0.attention.output.LayerNorm.bias    |        [768] |      768 |
| encoder.encoder.layer.0.intermediate.dense.weight          |  [3072, 768] |  2359296 |
| encoder.encoder.layer.0.intermediate.dense.bias            |       [3072] |     3072 |
| encoder.encoder.layer.0.output.dense.weight                |  [768, 3072] |  2359296 |
| encoder.encoder.layer.0.output.dense.bias                  |        [768] |      768 |
| encoder.encoder.layer.0.output.LayerNorm.weight            |        [768] |      768 |
| encoder.encoder.layer.0.output.LayerNorm.bias              |        [768] |      768 |
| encoder.encoder.layer.1.attention.self.query.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.1.attention.self.query.bias          |        [768] |      768 |
| encoder.encoder.layer.1.attention.self.key.weight          |   [768, 768] |   589824 |
| encoder.encoder.layer.1.attention.self.key.bias            |        [768] |      768 |
| encoder.encoder.layer.1.attention.self.value.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.1.attention.self.value.bias          |        [768] |      768 |
| encoder.encoder.layer.1.attention.output.dense.weight      |   [768, 768] |   589824 |
| encoder.encoder.layer.1.attention.output.dense.bias        |        [768] |      768 |
| encoder.encoder.layer.1.attention.output.LayerNorm.weight  |        [768] |      768 |
| encoder.encoder.layer.1.attention.output.LayerNorm.bias    |        [768] |      768 |
| encoder.encoder.layer.1.intermediate.dense.weight          |  [3072, 768] |  2359296 |
| encoder.encoder.layer.1.intermediate.dense.bias            |       [3072] |     3072 |
| encoder.encoder.layer.1.output.dense.weight                |  [768, 3072] |  2359296 |
| encoder.encoder.layer.1.output.dense.bias                  |        [768] |      768 |
| encoder.encoder.layer.1.output.LayerNorm.weight            |        [768] |      768 |
| encoder.encoder.layer.1.output.LayerNorm.bias              |        [768] |      768 |
| encoder.encoder.layer.2.attention.self.query.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.2.attention.self.query.bias          |        [768] |      768 |
| encoder.encoder.layer.2.attention.self.key.weight          |   [768, 768] |   589824 |
| encoder.encoder.layer.2.attention.self.key.bias            |        [768] |      768 |
| encoder.encoder.layer.2.attention.self.value.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.2.attention.self.value.bias          |        [768] |      768 |
| encoder.encoder.layer.2.attention.output.dense.weight      |   [768, 768] |   589824 |
| encoder.encoder.layer.2.attention.output.dense.bias        |        [768] |      768 |
| encoder.encoder.layer.2.attention.output.LayerNorm.weight  |        [768] |      768 |
| encoder.encoder.layer.2.attention.output.LayerNorm.bias    |        [768] |      768 |
| encoder.encoder.layer.2.intermediate.dense.weight          |  [3072, 768] |  2359296 |
| encoder.encoder.layer.2.intermediate.dense.bias            |       [3072] |     3072 |
| encoder.encoder.layer.2.output.dense.weight                |  [768, 3072] |  2359296 |
| encoder.encoder.layer.2.output.dense.bias                  |        [768] |      768 |
| encoder.encoder.layer.2.output.LayerNorm.weight            |        [768] |      768 |
| encoder.encoder.layer.2.output.LayerNorm.bias              |        [768] |      768 |
| encoder.encoder.layer.3.attention.self.query.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.3.attention.self.query.bias          |        [768] |      768 |
| encoder.encoder.layer.3.attention.self.key.weight          |   [768, 768] |   589824 |
| encoder.encoder.layer.3.attention.self.key.bias            |        [768] |      768 |
| encoder.encoder.layer.3.attention.self.value.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.3.attention.self.value.bias          |        [768] |      768 |
| encoder.encoder.layer.3.attention.output.dense.weight      |   [768, 768] |   589824 |
| encoder.encoder.layer.3.attention.output.dense.bias        |        [768] |      768 |
| encoder.encoder.layer.3.attention.output.LayerNorm.weight  |        [768] |      768 |
| encoder.encoder.layer.3.attention.output.LayerNorm.bias    |        [768] |      768 |
| encoder.encoder.layer.3.intermediate.dense.weight          |  [3072, 768] |  2359296 |
| encoder.encoder.layer.3.intermediate.dense.bias            |       [3072] |     3072 |
| encoder.encoder.layer.3.output.dense.weight                |  [768, 3072] |  2359296 |
| encoder.encoder.layer.3.output.dense.bias                  |        [768] |      768 |
| encoder.encoder.layer.3.output.LayerNorm.weight            |        [768] |      768 |
| encoder.encoder.layer.3.output.LayerNorm.bias              |        [768] |      768 |
| encoder.encoder.layer.4.attention.self.query.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.4.attention.self.query.bias          |        [768] |      768 |
| encoder.encoder.layer.4.attention.self.key.weight          |   [768, 768] |   589824 |
| encoder.encoder.layer.4.attention.self.key.bias            |        [768] |      768 |
| encoder.encoder.layer.4.attention.self.value.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.4.attention.self.value.bias          |        [768] |      768 |
| encoder.encoder.layer.4.attention.output.dense.weight      |   [768, 768] |   589824 |
| encoder.encoder.layer.4.attention.output.dense.bias        |        [768] |      768 |
| encoder.encoder.layer.4.attention.output.LayerNorm.weight  |        [768] |      768 |
| encoder.encoder.layer.4.attention.output.LayerNorm.bias    |        [768] |      768 |
| encoder.encoder.layer.4.intermediate.dense.weight          |  [3072, 768] |  2359296 |
| encoder.encoder.layer.4.intermediate.dense.bias            |       [3072] |     3072 |
| encoder.encoder.layer.4.output.dense.weight                |  [768, 3072] |  2359296 |
| encoder.encoder.layer.4.output.dense.bias                  |        [768] |      768 |
| encoder.encoder.layer.4.output.LayerNorm.weight            |        [768] |      768 |
| encoder.encoder.layer.4.output.LayerNorm.bias              |        [768] |      768 |
| encoder.encoder.layer.5.attention.self.query.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.5.attention.self.query.bias          |        [768] |      768 |
| encoder.encoder.layer.5.attention.self.key.weight          |   [768, 768] |   589824 |
| encoder.encoder.layer.5.attention.self.key.bias            |        [768] |      768 |
| encoder.encoder.layer.5.attention.self.value.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.5.attention.self.value.bias          |        [768] |      768 |
| encoder.encoder.layer.5.attention.output.dense.weight      |   [768, 768] |   589824 |
| encoder.encoder.layer.5.attention.output.dense.bias        |        [768] |      768 |
| encoder.encoder.layer.5.attention.output.LayerNorm.weight  |        [768] |      768 |
| encoder.encoder.layer.5.attention.output.LayerNorm.bias    |        [768] |      768 |
| encoder.encoder.layer.5.intermediate.dense.weight          |  [3072, 768] |  2359296 |
| encoder.encoder.layer.5.intermediate.dense.bias            |       [3072] |     3072 |
| encoder.encoder.layer.5.output.dense.weight                |  [768, 3072] |  2359296 |
| encoder.encoder.layer.5.output.dense.bias                  |        [768] |      768 |
| encoder.encoder.layer.5.output.LayerNorm.weight            |        [768] |      768 |
| encoder.encoder.layer.5.output.LayerNorm.bias              |        [768] |      768 |
| encoder.encoder.layer.6.attention.self.query.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.6.attention.self.query.bias          |        [768] |      768 |
| encoder.encoder.layer.6.attention.self.key.weight          |   [768, 768] |   589824 |
| encoder.encoder.layer.6.attention.self.key.bias            |        [768] |      768 |
| encoder.encoder.layer.6.attention.self.value.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.6.attention.self.value.bias          |        [768] |      768 |
| encoder.encoder.layer.6.attention.output.dense.weight      |   [768, 768] |   589824 |
| encoder.encoder.layer.6.attention.output.dense.bias        |        [768] |      768 |
| encoder.encoder.layer.6.attention.output.LayerNorm.weight  |        [768] |      768 |
| encoder.encoder.layer.6.attention.output.LayerNorm.bias    |        [768] |      768 |
| encoder.encoder.layer.6.intermediate.dense.weight          |  [3072, 768] |  2359296 |
| encoder.encoder.layer.6.intermediate.dense.bias            |       [3072] |     3072 |
| encoder.encoder.layer.6.output.dense.weight                |  [768, 3072] |  2359296 |
| encoder.encoder.layer.6.output.dense.bias                  |        [768] |      768 |
| encoder.encoder.layer.6.output.LayerNorm.weight            |        [768] |      768 |
| encoder.encoder.layer.6.output.LayerNorm.bias              |        [768] |      768 |
| encoder.encoder.layer.7.attention.self.query.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.7.attention.self.query.bias          |        [768] |      768 |
| encoder.encoder.layer.7.attention.self.key.weight          |   [768, 768] |   589824 |
| encoder.encoder.layer.7.attention.self.key.bias            |        [768] |      768 |
| encoder.encoder.layer.7.attention.self.value.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.7.attention.self.value.bias          |        [768] |      768 |
| encoder.encoder.layer.7.attention.output.dense.weight      |   [768, 768] |   589824 |
| encoder.encoder.layer.7.attention.output.dense.bias        |        [768] |      768 |
| encoder.encoder.layer.7.attention.output.LayerNorm.weight  |        [768] |      768 |
| encoder.encoder.layer.7.attention.output.LayerNorm.bias    |        [768] |      768 |
| encoder.encoder.layer.7.intermediate.dense.weight          |  [3072, 768] |  2359296 |
| encoder.encoder.layer.7.intermediate.dense.bias            |       [3072] |     3072 |
| encoder.encoder.layer.7.output.dense.weight                |  [768, 3072] |  2359296 |
| encoder.encoder.layer.7.output.dense.bias                  |        [768] |      768 |
| encoder.encoder.layer.7.output.LayerNorm.weight            |        [768] |      768 |
| encoder.encoder.layer.7.output.LayerNorm.bias              |        [768] |      768 |
| encoder.encoder.layer.8.attention.self.query.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.8.attention.self.query.bias          |        [768] |      768 |
| encoder.encoder.layer.8.attention.self.key.weight          |   [768, 768] |   589824 |
| encoder.encoder.layer.8.attention.self.key.bias            |        [768] |      768 |
| encoder.encoder.layer.8.attention.self.value.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.8.attention.self.value.bias          |        [768] |      768 |
| encoder.encoder.layer.8.attention.output.dense.weight      |   [768, 768] |   589824 |
| encoder.encoder.layer.8.attention.output.dense.bias        |        [768] |      768 |
| encoder.encoder.layer.8.attention.output.LayerNorm.weight  |        [768] |      768 |
| encoder.encoder.layer.8.attention.output.LayerNorm.bias    |        [768] |      768 |
| encoder.encoder.layer.8.intermediate.dense.weight          |  [3072, 768] |  2359296 |
| encoder.encoder.layer.8.intermediate.dense.bias            |       [3072] |     3072 |
| encoder.encoder.layer.8.output.dense.weight                |  [768, 3072] |  2359296 |
| encoder.encoder.layer.8.output.dense.bias                  |        [768] |      768 |
| encoder.encoder.layer.8.output.LayerNorm.weight            |        [768] |      768 |
| encoder.encoder.layer.8.output.LayerNorm.bias              |        [768] |      768 |
| encoder.encoder.layer.9.attention.self.query.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.9.attention.self.query.bias          |        [768] |      768 |
| encoder.encoder.layer.9.attention.self.key.weight          |   [768, 768] |   589824 |
| encoder.encoder.layer.9.attention.self.key.bias            |        [768] |      768 |
| encoder.encoder.layer.9.attention.self.value.weight        |   [768, 768] |   589824 |
| encoder.encoder.layer.9.attention.self.value.bias          |        [768] |      768 |
| encoder.encoder.layer.9.attention.output.dense.weight      |   [768, 768] |   589824 |
| encoder.encoder.layer.9.attention.output.dense.bias        |        [768] |      768 |
| encoder.encoder.layer.9.attention.output.LayerNorm.weight  |        [768] |      768 |
| encoder.encoder.layer.9.attention.output.LayerNorm.bias    |        [768] |      768 |
| encoder.encoder.layer.9.intermediate.dense.weight          |  [3072, 768] |  2359296 |
| encoder.encoder.layer.9.intermediate.dense.bias            |       [3072] |     3072 |
| encoder.encoder.layer.9.output.dense.weight                |  [768, 3072] |  2359296 |
| encoder.encoder.layer.9.output.dense.bias                  |        [768] |      768 |
| encoder.encoder.layer.9.output.LayerNorm.weight            |        [768] |      768 |
| encoder.encoder.layer.9.output.LayerNorm.bias              |        [768] |      768 |
| encoder.encoder.layer.10.attention.self.query.weight       |   [768, 768] |   589824 |
| encoder.encoder.layer.10.attention.self.query.bias         |        [768] |      768 |
| encoder.encoder.layer.10.attention.self.key.weight         |   [768, 768] |   589824 |
| encoder.encoder.layer.10.attention.self.key.bias           |        [768] |      768 |
| encoder.encoder.layer.10.attention.self.value.weight       |   [768, 768] |   589824 |
| encoder.encoder.layer.10.attention.self.value.bias         |        [768] |      768 |
| encoder.encoder.layer.10.attention.output.dense.weight     |   [768, 768] |   589824 |
| encoder.encoder.layer.10.attention.output.dense.bias       |        [768] |      768 |
| encoder.encoder.layer.10.attention.output.LayerNorm.weight |        [768] |      768 |
| encoder.encoder.layer.10.attention.output.LayerNorm.bias   |        [768] |      768 |
| encoder.encoder.layer.10.intermediate.dense.weight         |  [3072, 768] |  2359296 |
| encoder.encoder.layer.10.intermediate.dense.bias           |       [3072] |     3072 |
| encoder.encoder.layer.10.output.dense.weight               |  [768, 3072] |  2359296 |
| encoder.encoder.layer.10.output.dense.bias                 |        [768] |      768 |
| encoder.encoder.layer.10.output.LayerNorm.weight           |        [768] |      768 |
| encoder.encoder.layer.10.output.LayerNorm.bias             |        [768] |      768 |
| encoder.encoder.layer.11.attention.self.query.weight       |   [768, 768] |   589824 |
| encoder.encoder.layer.11.attention.self.query.bias         |        [768] |      768 |
| encoder.encoder.layer.11.attention.self.key.weight         |   [768, 768] |   589824 |
| encoder.encoder.layer.11.attention.self.key.bias           |        [768] |      768 |
| encoder.encoder.layer.11.attention.self.value.weight       |   [768, 768] |   589824 |
| encoder.encoder.layer.11.attention.self.value.bias         |        [768] |      768 |
| encoder.encoder.layer.11.attention.output.dense.weight     |   [768, 768] |   589824 |
| encoder.encoder.layer.11.attention.output.dense.bias       |        [768] |      768 |
| encoder.encoder.layer.11.attention.output.LayerNorm.weight |        [768] |      768 |
| encoder.encoder.layer.11.attention.output.LayerNorm.bias   |        [768] |      768 |
| encoder.encoder.layer.11.intermediate.dense.weight         |  [3072, 768] |  2359296 |
| encoder.encoder.layer.11.intermediate.dense.bias           |       [3072] |     3072 |
| encoder.encoder.layer.11.output.dense.weight               |  [768, 3072] |  2359296 |
| encoder.encoder.layer.11.output.dense.bias                 |        [768] |      768 |
| encoder.encoder.layer.11.output.LayerNorm.weight           |        [768] |      768 |
| encoder.encoder.layer.11.output.LayerNorm.bias             |        [768] |      768 |
| encoder.pooler.dense.weight                                |   [768, 768] |   589824 |
| encoder.pooler.dense.bias                                  |        [768] |      768 |
+------------------------------------------------------------+--------------+----------+
02/17/2024 13:45:12 - INFO - __main__ -   Training/evaluation parameters Namespace(agg_way='avg', aug_type_way='random_replace_type', code_length=256, codebase_file='dataset/Ruby/codebase.jsonl', config_name='DeepSoftwareAnalytics/CoCoSoDa', couninue_pre_train_data_files=['dataset/ruby/train.jsonl', 'dataset/java/train.jsonl'], data_aug_type='random_mask', data_flow_length=0, debug=False, device=device(type='cuda'), do_avg=False, do_continue_pre_trained=False, do_eval=False, do_fine_tune=False, do_ineer_loss=False, do_multi_lang_continue_pre_train=False, do_single_lang_continue_pre_train=False, do_test=True, do_train=True, do_whitening=False, do_zero_short=False, epoch=50, eval_batch_size=64, eval_data_file='dataset/Ruby/valid.jsonl', eval_frequency=100, fp16=False, gradient_accumulation_steps=1, hidden_size=768, lang='Ruby', learning_rate=2e-05, loaded_codebert_model_filename=None, loaded_model_filename=None, local_rank=-1, logging_steps=50, max_codeblock_num=10, max_grad_norm=1.0, max_steps=100, mlm_probability=0.1, mlp=False, moco_dim=768, moco_k=1024, moco_m=0.999, moco_t=0.07, moco_type='encoder_queue', model_name_or_path='DeepSoftwareAnalytics/CoCoSoDa', model_type='base', n_debug_samples=100, n_gpu=1, nl_length=128, num_train_epochs=5, num_warmup_steps=0, only_save_the_nl_code_vec=False, output_dir='./saved_models/fine_tune/Ruby', print_align_unif_loss=False, save_evaluation_reuslt=False, save_evaluation_reuslt_dir=None, save_steps=50, seed=123456, test_data_file='dataset/Ruby/test.jsonl', time_score=1, tokenizer_name='DeepSoftwareAnalytics/CoCoSoDa', train_batch_size=128, train_data_file='dataset/Ruby/train.jsonl', use_best_mrr_model=False, weight_decay=0.01)
Traceback (most recent call last):
  File "run.py", line 1188, in <module>
    main()
  File "run.py", line 1154, in main
    train(args, model, tokenizer, pool)
  File "run.py", line 534, in train
    train_dataset=TextDataset_unixcoder(tokenizer, args, args.train_data_file, pool)
  File "run.py", line 393, in __init__
    with open(file_path) as f:
FileNotFoundError: [Errno 2] No such file or directory: 'dataset/Ruby/train.jsonl'