google-research / google-research/language
about orqa model running
- Dominant language
- Python
- Stars
- 1.8k
- Forks
- 362
- PR merge metrics
- No merged PRs in 30d
Description
hi i have changed the following command to running on GPU device
Training on TPU
MODEL_DIR=gs:///
TFHUB_CACHE_DIR=gs:///
TFHUB_CACHE_DIR=$TFHUB_CACHE_DIR \
TPU_NAME=
python -m language.orqa.experiments.ict_experiment \
--model_dir=$MODEL_DIR \
--bert_hub_module_path=https://tfhub.dev/google/bert_uncased_L-12_H-768_A-12/1 \
--examples_path=gs://orqa-data/enwiki-20181220/examples.tfr \
--save_checkpoints_steps=1000 \
--batch_size=4096 \
--num_train_steps=100000 \
--tpu_name=$TPU_NAME \
--use_tpu=True
to like this :
python -m language.orqa.experiments.ict_experiment \
--model_dir=$MODEL_DIR \
--bert_hub_module_path=https://tfhub.dev/google/bert_uncased_L-12_H-768_A-12/1 \
--examples_path=gs://orqa-data/enwiki-20181220/examples.tfr \
--save_checkpoints_steps=1000 \
--batch_size=4096 \
--num_train_steps=100000 \
--use_tpu=False
batchsize is about 256
but it seems too slow 100 steps needs 8409.323 sec i have 4 tesla v100 GPUS i donot know whether config right for this running .
Thank you
Contributor guide
Assessment
This issue has not been assessed yet.