Skip to content

Commit 5f2b14e

Browse files
committed
revise the changes please
1 parent baf6e6e commit 5f2b14e

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

TensorFlow/LanguageModeling/BERT/trtis/run_squad_trtis_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def append_feature(feature):
121121
protocol_str = 'grpc' # http or grpc
122122
url = FLAGS.trtis_server_url
123123
verbose = True
124-
model_name = FLAGS.trtis_model_name
124+
model_name = 'resnet50_netdef' #FLAGS.trtis_model_name
125125
model_version = FLAGS.trtis_model_version
126126
batch_size = FLAGS.predict_batch_size
127127

TensorFlow/LanguageModeling/BERT/trtis/scripts/run_trtis.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ trtis_model_overwrite=${14:-"False"}
3030

3131
if [ "$bert_model" = "large" ] ; then
3232
# export BERT_DIR=data/download/google_pretrained_weights/uncased_L-24_H-1024_A-16
33-
export BERT_DIR=data/download/google_pretrained_weights/wwm_uncased_L-24_H-1024_A-16
33+
export BERT_DIR=/opt/luciapp/apps/service/DeepLearningExamples/TensorFlow/LanguageModeling/BERT/data/download/google_pretrained_weights/uncased_L-24_H-1024_A-16
3434
else
3535
export BERT_DIR=data/download/google_pretrained_weights/uncased_L-12_H-768_A-12
3636
fi

0 commit comments

Comments
 (0)