diff --git "a/log/train_log" "b/log/train_log" new file mode 100644--- /dev/null +++ "b/log/train_log" @@ -0,0 +1,13043 @@ +# Running on vgnj007 +# Started at Thu 10 Nov 16:02:29 CET 2022 +# python3 src/run_speech_recognition_ctc.py --report_to=none --run_name=experiments/results/robust_model/wav2vec2-large-960h-lv60-self/atcosim_corpus/0.0ld_0.05ad_0.05attd_0.0fpd_0.03mtp_10mtl_0.0mfp_10mfl --preprocessing_num_workers=5 --model_name_or_path=facebook/wav2vec2-large-960h-lv60-self --dataset_name=experiments/data/atcosim_corpus/train --min_duration_in_seconds=1.0 --max_duration_in_seconds=20 --eval_dataset_name=experiments/data/atcosim_corpus/test --train_split_name=train --output_dir=experiments/results/robust_model/wav2vec2-large-960h-lv60-self/atcosim_corpus/0.0ld_0.05ad_0.05attd_0.0fpd_0.03mtp_10mtl_0.0mfp_10mfl --num_train_epochs=50 --per_device_train_batch_size=24 --per_device_eval_batch_size=24 --gradient_accumulation_steps=4 --learning_rate=5e-4 --weight_decay=0.001 --warmup_steps=500 --evaluation_strategy=steps --text_column_name=text --audio_column_name=audio --length_column_name=input_length '--chars_to_ignore=, ? . ! \; \: " “ % ‘ ” � — ’ … –' --save_steps=500 --eval_steps=500 --logging_steps=500 --layerdrop=0.0 --activation_dropout=0.05 --attention_dropout=0.05 --save_total_limit=1 --feat_proj_dropout=0.0 --mask_time_prob=0.03 --mask_time_length=10 --mask_feature_prob=0.0 --mask_feature_length=10 --gradient_checkpointing --freeze_feature_encoder --fp16 --group_by_length --do_train --do_eval --max_steps 20000 --overwrite_output_dir --freeze_feature_encoder +11/10/2022 16:02:41 - WARNING - __main__ - Process rank: -1, device: cuda:0, n_gpu: 1distributed training: False, 16-bits training: True +11/10/2022 16:02:41 - INFO - __main__ - Training/evaluation parameters TrainingArguments( +_n_gpu=1, +adafactor=False, +adam_beta1=0.9, +adam_beta2=0.999, +adam_epsilon=1e-08, +auto_find_batch_size=False, +bf16=False, +bf16_full_eval=False, +data_seed=None, +dataloader_drop_last=False, +dataloader_num_workers=0, +dataloader_pin_memory=True, +ddp_bucket_cap_mb=None, +ddp_find_unused_parameters=None, +ddp_timeout=1800, +debug=[], +deepspeed=None, +disable_tqdm=False, +do_eval=True, +do_predict=False, +do_train=True, +eval_accumulation_steps=None, +eval_delay=0, +eval_steps=500, +evaluation_strategy=steps, +fp16=True, +fp16_backend=auto, +fp16_full_eval=False, +fp16_opt_level=O1, +fsdp=[], +fsdp_min_num_params=0, +fsdp_transformer_layer_cls_to_wrap=None, +full_determinism=False, +gradient_accumulation_steps=4, +gradient_checkpointing=True, +greater_is_better=None, +group_by_length=True, +half_precision_backend=auto, +hub_model_id=None, +hub_private_repo=False, +hub_strategy=every_save, +hub_token=, +ignore_data_skip=False, +include_inputs_for_metrics=False, +jit_mode_eval=False, +label_names=None, +label_smoothing_factor=0.0, +learning_rate=0.0005, +length_column_name=input_length, +load_best_model_at_end=False, +local_rank=-1, +log_level=passive, +log_level_replica=passive, +log_on_each_node=True, +logging_dir=experiments/results/robust_model/wav2vec2-large-960h-lv60-self/atcosim_corpus/0.0ld_0.05ad_0.05attd_0.0fpd_0.03mtp_10mtl_0.0mfp_10mfl/runs/Nov10_16-02-41_vgnj007, +logging_first_step=False, +logging_nan_inf_filter=True, +logging_steps=500, +logging_strategy=steps, +lr_scheduler_type=linear, +max_grad_norm=1.0, +max_steps=20000, +metric_for_best_model=None, +mp_parameters=, +no_cuda=False, +num_train_epochs=50.0, +optim=adamw_hf, +output_dir=experiments/results/robust_model/wav2vec2-large-960h-lv60-self/atcosim_corpus/0.0ld_0.05ad_0.05attd_0.0fpd_0.03mtp_10mtl_0.0mfp_10mfl, +overwrite_output_dir=True, +past_index=-1, +per_device_eval_batch_size=24, +per_device_train_batch_size=24, +prediction_loss_only=False, +push_to_hub=False, +push_to_hub_model_id=None, +push_to_hub_organization=None, +push_to_hub_token=, +ray_scope=last, +remove_unused_columns=True, +report_to=[], +resume_from_checkpoint=None, +run_name=experiments/results/robust_model/wav2vec2-large-960h-lv60-self/atcosim_corpus/0.0ld_0.05ad_0.05attd_0.0fpd_0.03mtp_10mtl_0.0mfp_10mfl, +save_on_each_node=False, +save_steps=500, +save_strategy=steps, +save_total_limit=1, +seed=42, +sharded_ddp=[], +skip_memory_metrics=True, +tf32=None, +torchdynamo=None, +tpu_metrics_debug=False, +tpu_num_cores=None, +use_ipex=False, +use_legacy_prediction_loop=False, +use_mps_device=False, +warmup_ratio=0.0, +warmup_steps=500, +weight_decay=0.001, +xpu_backend=None, +) +11/10/2022 16:02:41 - WARNING - datasets.builder - Using custom data configuration train-982961f3786896dd +11/10/2022 16:02:41 - WARNING - datasets.builder - Found cached dataset atc_data_loader (/remote/idiap.svm/temp.speech01/jzuluaga/experiments/journal/asr/github/w2v2-air-traffic/experiments/data/atcosim_corpus/train/cache/atc_data_loader/train-982961f3786896dd/0.0.0/3dc36b96f5a9e12dd97c69117ef840e3e971dd43ca1b0750cf000d34695b9d81) +11/10/2022 16:02:42 - WARNING - datasets.builder - Using custom data configuration test-24d2db3884146b91 +11/10/2022 16:02:42 - WARNING - datasets.builder - Found cached dataset atc_data_loader (/remote/idiap.svm/temp.speech01/jzuluaga/experiments/journal/asr/github/w2v2-air-traffic/experiments/data/atcosim_corpus/test/cache/atc_data_loader/test-24d2db3884146b91/0.0.0/3dc36b96f5a9e12dd97c69117ef840e3e971dd43ca1b0750cf000d34695b9d81) +loading configuration file config.json from cache at /idiap/temp/jzuluaga/cache/huggingface/models--facebook--wav2vec2-large-960h-lv60-self/snapshots/54074b1c16f4de6a5ad59affb4caa8f2ea03a119/config.json +Model config Wav2Vec2Config { + "_name_or_path": "facebook/wav2vec2-large-960h-lv60-self", + "activation_dropout": 0.1, + "adapter_kernel_size": 3, + "adapter_stride": 2, + "add_adapter": false, + "apply_spec_augment": true, + "architectures": [ + "Wav2Vec2ForCTC" + ], + "attention_dropout": 0.1, + "bos_token_id": 1, + "classifier_proj_size": 256, + "codevector_dim": 256, + "contrastive_logits_temperature": 0.1, + "conv_bias": true, + "conv_dim": [ + 512, + 512, + 512, + 512, + 512, + 512, + 512 + ], + "conv_kernel": [ + 10, + 3, + 3, + 3, + 3, + 2, + 2 + ], + "conv_stride": [ + 5, + 2, + 2, + 2, + 2, + 2, + 2 + ], + "ctc_loss_reduction": "sum", + "ctc_zero_infinity": false, + "diversity_loss_weight": 0.1, + "do_stable_layer_norm": true, + "eos_token_id": 2, + "feat_extract_activation": "gelu", + "feat_extract_dropout": 0.0, + "feat_extract_norm": "layer", + "feat_proj_dropout": 0.1, + "feat_quantizer_dropout": 0.0, + "final_dropout": 0.1, + "gradient_checkpointing": false, + "hidden_act": "gelu", + "hidden_dropout": 0.1, + "hidden_dropout_prob": 0.1, + "hidden_size": 1024, + "initializer_range": 0.02, + "intermediate_size": 4096, + "layer_norm_eps": 1e-05, + "layerdrop": 0.1, + "mask_feature_length": 10, + "mask_feature_min_masks": 0, + "mask_feature_prob": 0.0, + "mask_time_length": 10, + "mask_time_min_masks": 2, + "mask_time_prob": 0.05, + "model_type": "wav2vec2", + "num_adapter_layers": 3, + "num_attention_heads": 16, + "num_codevector_groups": 2, + "num_codevectors_per_group": 320, + "num_conv_pos_embedding_groups": 16, + "num_conv_pos_embeddings": 128, + "num_feat_extract_layers": 7, + "num_hidden_layers": 24, + "num_negatives": 100, + "output_hidden_size": 1024, + "pad_token_id": 0, + "proj_codevector_dim": 256, + "tdnn_dilation": [ + 1, + 2, + 3, + 1, + 1 + ], + "tdnn_dim": [ + 512, + 512, + 512, + 512, + 1500 + ], + "tdnn_kernel": [ + 5, + 3, + 3, + 1, + 1 + ], + "transformers_version": "4.24.0", + "use_weighted_layer_sum": false, + "vocab_size": 32, + "xvector_output_dim": 512 +} + + 0%| | 0/1 [00:00 to the vocabulary +Adding to the vocabulary +Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained. +loading configuration file preprocessor_config.json from cache at /idiap/temp/jzuluaga/cache/huggingface/models--facebook--wav2vec2-large-960h-lv60-self/snapshots/54074b1c16f4de6a5ad59affb4caa8f2ea03a119/preprocessor_config.json +loading configuration file config.json from cache at /idiap/temp/jzuluaga/cache/huggingface/models--facebook--wav2vec2-large-960h-lv60-self/snapshots/54074b1c16f4de6a5ad59affb4caa8f2ea03a119/config.json +Model config Wav2Vec2Config { + "_name_or_path": "facebook/wav2vec2-large-960h-lv60-self", + "activation_dropout": 0.1, + "adapter_kernel_size": 3, + "adapter_stride": 2, + "add_adapter": false, + "apply_spec_augment": true, + "architectures": [ + "Wav2Vec2ForCTC" + ], + "attention_dropout": 0.1, + "bos_token_id": 1, + "classifier_proj_size": 256, + "codevector_dim": 256, + "contrastive_logits_temperature": 0.1, + "conv_bias": true, + "conv_dim": [ + 512, + 512, + 512, + 512, + 512, + 512, + 512 + ], + "conv_kernel": [ + 10, + 3, + 3, + 3, + 3, + 2, + 2 + ], + "conv_stride": [ + 5, + 2, + 2, + 2, + 2, + 2, + 2 + ], + "ctc_loss_reduction": "sum", + "ctc_zero_infinity": false, + "diversity_loss_weight": 0.1, + "do_stable_layer_norm": true, + "eos_token_id": 2, + "feat_extract_activation": "gelu", + "feat_extract_dropout": 0.0, + "feat_extract_norm": "layer", + "feat_proj_dropout": 0.1, + "feat_quantizer_dropout": 0.0, + "final_dropout": 0.1, + "gradient_checkpointing": false, + "hidden_act": "gelu", + "hidden_dropout": 0.1, + "hidden_dropout_prob": 0.1, + "hidden_size": 1024, + "initializer_range": 0.02, + "intermediate_size": 4096, + "layer_norm_eps": 1e-05, + "layerdrop": 0.1, + "mask_feature_length": 10, + "mask_feature_min_masks": 0, + "mask_feature_prob": 0.0, + "mask_time_length": 10, + "mask_time_min_masks": 2, + "mask_time_prob": 0.05, + "model_type": "wav2vec2", + "num_adapter_layers": 3, + "num_attention_heads": 16, + "num_codevector_groups": 2, + "num_codevectors_per_group": 320, + "num_conv_pos_embedding_groups": 16, + "num_conv_pos_embeddings": 128, + "num_feat_extract_layers": 7, + "num_hidden_layers": 24, + "num_negatives": 100, + "output_hidden_size": 1024, + "pad_token_id": 0, + "proj_codevector_dim": 256, + "tdnn_dilation": [ + 1, + 2, + 3, + 1, + 1 + ], + "tdnn_dim": [ + 512, + 512, + 512, + 512, + 1500 + ], + "tdnn_kernel": [ + 5, + 3, + 3, + 1, + 1 + ], + "transformers_version": "4.24.0", + "use_weighted_layer_sum": false, + "vocab_size": 32, + "xvector_output_dim": 512 +} + +Feature extractor Wav2Vec2FeatureExtractor { + "do_normalize": true, + "feature_extractor_type": "Wav2Vec2FeatureExtractor", + "feature_size": 1, + "padding_side": "right", + "padding_value": 0.0, + "return_attention_mask": true, + "sampling_rate": 16000 +} + +loading weights file pytorch_model.bin from cache at /idiap/temp/jzuluaga/cache/huggingface/models--facebook--wav2vec2-large-960h-lv60-self/snapshots/54074b1c16f4de6a5ad59affb4caa8f2ea03a119/pytorch_model.bin +All model checkpoint weights were used when initializing Wav2Vec2ForCTC. + +Some weights of Wav2Vec2ForCTC were not initialized from the model checkpoint at facebook/wav2vec2-large-960h-lv60-self and are newly initialized: ['wav2vec2.masked_spec_embed'] +You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference. +Some weights of Wav2Vec2ForCTC were not initialized from the model checkpoint at facebook/wav2vec2-large-960h-lv60-self and are newly initialized because the shapes did not match: +- lm_head.weight: found shape torch.Size([32, 1024]) in the checkpoint and torch.Size([31, 1024]) in the model instantiated +- lm_head.bias: found shape torch.Size([32]) in the checkpoint and torch.Size([31]) in the model instantiated +You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference. + preprocess datasets #0: 0%| | 0/1528 [00:00 to the vocabulary +Adding to the vocabulary +max_steps is given, it will override any value given in num_train_epochs +Using cuda_amp half precision backend +The following columns in the training set don't have a corresponding argument in `Wav2Vec2ForCTC.forward` and have been ignored: input_length. If input_length are not expected by `Wav2Vec2ForCTC.forward`, you can safely ignore this message. +/idiap/home/jzuluaga/.local/lib/python3.9/site-packages/transformers/optimization.py:306: FutureWarning: This implementation of AdamW is deprecated and will be removed in a future version. Use the PyTorch implementation torch.optim.AdamW instead, or set `no_deprecation_warning=True` to disable this warning + warnings.warn( +***** Running training ***** + Num examples = 7480 + Num Epochs = 257 + Instantaneous batch size per device = 24 + Total train batch size (w. parallel, distributed & accumulation) = 96 + Gradient Accumulation steps = 4 + Total optimization steps = 20000 + Number of trainable parameters = 311260319 + 0%| | 0/20000 [00:00