flax-community
/

Bengali-t5

Text2Text Generation

Inference Endpoints

Model card Files Files and versions Community

ibraheemmoosa commited on Jul 15, 2021

Commit

77a2c93

•

1 Parent(s): 062d2af

Change model directory to .

Files changed (1) hide show

t5-train.sh +2 -2

t5-train.sh CHANGED Viewed

@@ -1,9 +1,9 @@
 #!/bin/bash
 HF_DATASETS_IN_MEMORY_MAX_SIZE=200000000
-MODEL_DIR=checkpoints
 TOKENIZER_DIR=tokenizer
 ./run_t5_mlm_flax.py \
-    --output_dir="${MODEL_DIR}" \
     --model_type="t5" \
     --config_name="t5-small" \
     --tokenizer_name="${TOKENIZER_DIR}" \

 #!/bin/bash
 HF_DATASETS_IN_MEMORY_MAX_SIZE=200000000
+MODEL_DIR=.
 TOKENIZER_DIR=tokenizer
 ./run_t5_mlm_flax.py \
+    --output_dir="${MODEL_DIR}/checkpoints" \
     --model_type="t5" \
     --config_name="t5-small" \
     --tokenizer_name="${TOKENIZER_DIR}" \