sanchit-gandhi's picture
2hx8pk65: saving weights and logs of step 10k
f74be82
raw
history blame
No virus
2.23 kB
{
"os": "Linux-5.11.0-1028-gcp-x86_64-with-glibc2.29",
"python": "3.8.10",
"heartbeatAt": "2022-08-28T08:44:08.435675",
"startedAt": "2022-08-28T08:44:07.234991",
"docker": null,
"cpu_count": 96,
"cuda": null,
"args": [
"--dataset_name=librispeech_asr",
"--model_name_or_path=sanchit-gandhi/flax-wav2vec2-2-bart-large-scan",
"--dataset_config_name=all",
"--train_split_name=train.clean.100+train.clean.360+train.other.500",
"--eval_split_name=validation.clean",
"--test_split_name=validation.other+test.clean+test.other",
"--text_column_name=text",
"--id_column_name=id",
"--output_dir=./",
"--wandb_project=librispeech_960h",
"--wandb_name=flax-wav2vec2-2-bart-large-ls-960h-black-box",
"--dataset_cache_dir=/home/sanchitgandhi/cache/huggingface/datasets",
"--per_device_train_batch_size=8",
"--per_device_eval_batch_size=4",
"--learning_rate=1e-4",
"--warmup_steps=500",
"--logging_steps=25",
"--max_steps=50000",
"--eval_steps=10000",
"--save_steps=10000",
"--generation_max_length=200",
"--generation_num_beams=5",
"--generation_length_penalty=1.2",
"--hidden_dropout=0.2",
"--activation_dropout=0.2",
"--feat_proj_dropout=0.2",
"--overwrite_output_dir",
"--gradient_checkpointing",
"--freeze_feature_encoder",
"--predict_with_generate",
"--do_lower_case",
"--do_eval",
"--do_train",
"--do_predict",
"--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-ls-960h-black-box",
"commit": "140399a622e2a82685fa4b9727f3d970b8bef9e0"
},
"email": "sanchit@huggingface.co",
"root": "/home/sanchitgandhi/flax-wav2vec2-2-bart-large-ls-960h-black-box",
"host": "t1v-n-5966b949-w-0",
"username": "sanchitgandhi",
"executable": "/home/sanchitgandhi/hf/bin/python"
}