sanchit-gandhi's picture
Saving weights and logs of epoch 0
4f730f1
{
"os": "Linux-5.4.0-1043-gcp-x86_64-with-glibc2.29",
"python": "3.8.10",
"heartbeatAt": "2022-03-23T18:02:14.291985",
"startedAt": "2022-03-23T18:02:11.053380",
"docker": null,
"cpu_count": 96,
"cuda": null,
"args": [
"--dataset_name=librispeech_asr",
"--model_name_or_path=./",
"--dataset_config_name=clean",
"--train_split_name=train.100",
"--eval_split_name=validation",
"--output_dir=./",
"--dataset_cache_dir=/home/sanchit_huggingface_co/cache/huggingface/datasets",
"--cache_dir=/home/sanchit_huggingface_co/cache/huggingface/transformers",
"--preprocessing_num_workers=16",
"--length_column_name=input_length",
"--overwrite_output_dir",
"--num_train_epochs=5",
"--per_device_train_batch_size=2",
"--per_device_eval_batch_size=2",
"--gradient_accumulation_steps=8",
"--logging_steps=25",
"--max_duration_in_seconds=15",
"--max_target_length=64",
"--generation_max_length=40",
"--generation_num_beams=1",
"--learning_rate=3e-4",
"--warmup_steps=500",
"--text_column_name=text",
"--save_total_limit=1",
"--freeze_feature_encoder",
"--predict_with_generate",
"--do_lower_case",
"--do_eval",
"--do_train",
"--push_to_hub",
"--use_auth_token"
],
"state": "running",
"program": "run_flax_speech_recognition_seq2seq.py",
"codePath": "run_flax_speech_recognition_seq2seq.py",
"git": {
"remote": "https://huggingface.co/sanchit-gandhi/flax-wav2vec2-2-bart-large-cnn-gradient-accumulation",
"commit": "d1558b4bad2032d0c74efadebceaa6f07bb6393c"
},
"email": "sanchit@huggingface.co",
"root": "/home/sanchit_huggingface_co/flax-wav2vec2-2-bart-large-cnn-gradient-accumulation",
"host": "t1v-n-0e948115-w-0",
"username": "sanchit_huggingface_co",
"executable": "/home/sanchit_huggingface_co/venv/bin/python"
}