"mmdet3d/models/layers/sparse_block.py" did not exist on "ed81c6cc326a33f927d4455c219bb7715591b08b"
bert_class.sh 670 Bytes
Newer Older
hepj987's avatar
hepj987 committed
1
2
3
4
5
6
7
8
9
10
export HIP_VISIBLE_DEVICES=0
python3 run_classifier.py \
  --mode train_and_eval \
  --input_meta_data_path //public/home/hepj/MNLI/meta_data \
  --train_data_path /public/home/hepj/MNLI/train.tf_record \
  --eval_data_path /public/home/hepj/MNLI/eval.tf_record \
  --bert_config_file /public/home/hepj/model_source/uncased_L-12_H-768_A-12/bert_config.json \
  --init_checkpoint /public/home/hepj/model_source/bert-base-TF2/bert_model.ckpt \
  --train_batch_size  320 \
  --eval_batch_size 32 \
hepj987's avatar
hepj987 committed
11
  --steps_per_loop 10 \
hepj987's avatar
hepj987 committed
12
13
14
15
16
  --learning_rate 2e-5 \
  --num_train_epochs 3 \
  --num_gpus 1 \
  --model_dir /public/home/hepj/model/tf2/out1 \
  --distribution_strategy mirrored