2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_setup.py:_flush():76] Current SDK version is 0.16.1 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_setup.py:_flush():76] Configure stats pid to 122884 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_setup.py:_flush():76] Loading settings from /home/sanchit/.config/wandb/settings 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_setup.py:_flush():76] Loading settings from /home/sanchit/whisper-large-v3-ft-cv16-mn-freeze-encoder/wandb/settings 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_setup.py:_flush():76] Loading settings from environment variables: {} 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_setup.py:_flush():76] Applying setup settings: {'_disable_service': False} 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_setup.py:_flush():76] Inferring run settings from compute environment: {'program_relpath': 'run_speech_recognition_seq2seq.py', 'program_abspath': '/home/sanchit/whisper-large-v3-ft-cv16-mn-freeze-encoder/run_speech_recognition_seq2seq.py', 'program': 'run_speech_recognition_seq2seq.py'} 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_init.py:_log_setup():524] Logging user logs to /home/sanchit/whisper-large-v3-ft-cv16-mn-freeze-encoder/wandb/run-20240122_192258-yf2elmz6/logs/debug.log 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_init.py:_log_setup():525] Logging internal logs to /home/sanchit/whisper-large-v3-ft-cv16-mn-freeze-encoder/wandb/run-20240122_192258-yf2elmz6/logs/debug-internal.log 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_init.py:init():564] calling init triggers 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_init.py:init():571] wandb.init called with sweep_config: {} config: {} 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_init.py:init():614] starting backend 2024-01-22 19:22:58,027 INFO MainThread:122884 [wandb_init.py:init():618] setting up manager 2024-01-22 19:22:58,028 INFO MainThread:122884 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn 2024-01-22 19:22:58,029 INFO MainThread:122884 [wandb_init.py:init():624] backend started and connected 2024-01-22 19:22:58,032 INFO MainThread:122884 [wandb_init.py:init():716] updated telemetry 2024-01-22 19:22:58,152 INFO MainThread:122884 [wandb_init.py:init():749] communicating run to backend with 90.0 second timeout 2024-01-22 19:22:58,490 INFO MainThread:122884 [wandb_run.py:_on_init():2254] communicating current version 2024-01-22 19:22:58,557 INFO MainThread:122884 [wandb_run.py:_on_init():2263] got version response upgrade_message: "wandb version 0.16.2 is available! To upgrade, please run:\n $ pip install wandb --upgrade" 2024-01-22 19:22:58,557 INFO MainThread:122884 [wandb_init.py:init():800] starting run threads in backend 2024-01-22 19:22:58,634 INFO MainThread:122884 [wandb_run.py:_console_start():2233] atexit reg 2024-01-22 19:22:58,634 INFO MainThread:122884 [wandb_run.py:_redirect():2088] redirect: wrap_raw 2024-01-22 19:22:58,634 INFO MainThread:122884 [wandb_run.py:_redirect():2153] Wrapping output streams. 2024-01-22 19:22:58,634 INFO MainThread:122884 [wandb_run.py:_redirect():2178] Redirects installed. 2024-01-22 19:22:58,635 INFO MainThread:122884 [wandb_init.py:init():841] run started, returning control to user process 2024-01-22 19:22:58,637 INFO MainThread:122884 [wandb_run.py:_config_callback():1342] config_cb None None {'vocab_size': 51866, 'num_mel_bins': 128, 'd_model': 1280, 'encoder_layers': 32, 'encoder_attention_heads': 20, 'decoder_layers': 32, 'decoder_attention_heads': 20, 'decoder_ffn_dim': 5120, 'encoder_ffn_dim': 5120, 'dropout': 0.0, 'attention_dropout': 0.0, 'activation_dropout': 0.0, 'activation_function': 'gelu', 'init_std': 0.02, 'encoder_layerdrop': 0.0, 'decoder_layerdrop': 0.0, 'use_cache': False, 'num_hidden_layers': 32, 'scale_embedding': False, 'max_source_positions': 1500, 'max_target_positions': 448, 'classifier_proj_size': 256, 'use_weighted_layer_sum': False, 'apply_spec_augment': False, 'mask_time_prob': 0.05, 'mask_time_length': 10, 'mask_time_min_masks': 2, 'mask_feature_prob': 0.0, 'mask_feature_length': 10, 'mask_feature_min_masks': 0, 'median_filter_width': 7, 'return_dict': True, 'output_hidden_states': False, 'output_attentions': False, 'torchscript': False, 'torch_dtype': 'float16', 'use_bfloat16': False, 'tf_legacy_loss': False, 'pruned_heads': {}, 'tie_word_embeddings': True, 'chunk_size_feed_forward': 0, 'is_encoder_decoder': True, 'is_decoder': False, 'cross_attention_hidden_size': None, 'add_cross_attention': False, 'tie_encoder_decoder': False, 'max_length': 448, 'min_length': 0, 'do_sample': False, 'early_stopping': False, 'num_beams': 1, 'num_beam_groups': 1, 'diversity_penalty': 0.0, 'temperature': 1.0, 'top_k': 50, 'top_p': 1.0, 'typical_p': 1.0, 'repetition_penalty': 1.0, 'length_penalty': 1.0, 'no_repeat_ngram_size': 0, 'encoder_no_repeat_ngram_size': 0, 'bad_words_ids': None, 'num_return_sequences': 1, 'output_scores': False, 'return_dict_in_generate': False, 'forced_bos_token_id': None, 'forced_eos_token_id': None, 'remove_invalid_values': False, 'exponential_decay_length_penalty': None, 'suppress_tokens': None, 'begin_suppress_tokens': [220, 50257], 'architectures': ['WhisperForConditionalGeneration'], 'finetuning_task': None, 'id2label': {0: 'LABEL_0', 1: 'LABEL_1'}, 'label2id': {'LABEL_0': 0, 'LABEL_1': 1}, 'tokenizer_class': None, 'prefix': None, 'bos_token_id': 50257, 'pad_token_id': 50256, 'eos_token_id': 50257, 'sep_token_id': None, 'decoder_start_token_id': 50258, 'task_specific_params': None, 'problem_type': None, '_name_or_path': 'openai/whisper-large-v3', 'transformers_version': '4.38.0.dev0', 'model_type': 'whisper', 'forced_decoder_ids': None, 'output_dir': './', 'overwrite_output_dir': True, 'do_train': True, 'do_eval': True, 'do_predict': False, 'evaluation_strategy': 'steps', 'prediction_loss_only': False, 'per_device_train_batch_size': 32, 'per_device_eval_batch_size': 32, 'per_gpu_train_batch_size': None, 'per_gpu_eval_batch_size': None, 'gradient_accumulation_steps': 1, 'eval_accumulation_steps': None, 'eval_delay': 0, 'learning_rate': 0.0001, 'weight_decay': 0.0, 'adam_beta1': 0.9, 'adam_beta2': 0.999, 'adam_epsilon': 1e-08, 'max_grad_norm': 1.0, 'num_train_epochs': 10.0, 'max_steps': -1, 'lr_scheduler_type': 'linear', 'lr_scheduler_kwargs': {}, 'warmup_ratio': 0.0, 'warmup_steps': 500, 'log_level': 'passive', 'log_level_replica': 'warning', 'log_on_each_node': True, 'logging_dir': './runs/Jan22_19-21-13_hf-dgx-01', 'logging_strategy': 'steps', 'logging_first_step': False, 'logging_steps': 25, 'logging_nan_inf_filter': True, 'save_strategy': 'steps', 'save_steps': 1000, 'save_total_limit': None, 'save_safetensors': True, 'save_on_each_node': False, 'save_only_model': False, 'no_cuda': False, 'use_cpu': False, 'use_mps_device': False, 'seed': 42, 'data_seed': None, 'jit_mode_eval': False, 'use_ipex': False, 'bf16': False, 'fp16': True, 'fp16_opt_level': 'O1', 'half_precision_backend': 'auto', 'bf16_full_eval': False, 'fp16_full_eval': True, 'tf32': None, 'local_rank': 0, 'ddp_backend': None, 'tpu_num_cores': None, 'tpu_metrics_debug': False, 'debug': [], 'dataloader_drop_last': False, 'eval_steps': 500, 'dataloader_num_workers': 0, 'past_index': -1, 'run_name': './', 'disable_tqdm': False, 'remove_unused_columns': True, 'label_names': None, 'load_best_model_at_end': False, 'metric_for_best_model': None, 'greater_is_better': None, 'ignore_data_skip': False, 'fsdp': [], 'fsdp_min_num_params': 0, 'fsdp_config': {'min_num_params': 0, 'xla': False, 'xla_fsdp_grad_ckpt': False}, 'fsdp_transformer_layer_cls_to_wrap': None, 'deepspeed': None, 'label_smoothing_factor': 0.0, 'optim': 'adamw_torch', 'optim_args': None, 'adafactor': False, 'group_by_length': False, 'length_column_name': 'input_length', 'report_to': ['tensorboard', 'wandb'], 'ddp_find_unused_parameters': None, 'ddp_bucket_cap_mb': None, 'ddp_broadcast_buffers': None, 'dataloader_pin_memory': True, 'dataloader_persistent_workers': False, 'skip_memory_metrics': True, 'use_legacy_prediction_loop': False, 'push_to_hub': True, 'resume_from_checkpoint': None, 'hub_model_id': None, 'hub_strategy': 'every_save', 'hub_token': '', 'hub_private_repo': False, 'hub_always_push': False, 'gradient_checkpointing': True, 'gradient_checkpointing_kwargs': None, 'include_inputs_for_metrics': False, 'fp16_backend': 'auto', 'push_to_hub_model_id': None, 'push_to_hub_organization': None, 'push_to_hub_token': '', 'mp_parameters': '', 'auto_find_batch_size': False, 'full_determinism': False, 'torchdynamo': None, 'ray_scope': 'last', 'ddp_timeout': 1800, 'torch_compile': False, 'torch_compile_backend': None, 'torch_compile_mode': None, 'dispatch_batches': None, 'split_batches': False, 'include_tokens_per_second': False, 'include_num_input_tokens_seen': False, 'neftune_noise_alpha': None, 'sortish_sampler': False, 'predict_with_generate': True, 'generation_max_length': 225, 'generation_num_beams': None, 'generation_config': None}