File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
TensorFlow/LanguageModeling/BERT/trtis Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change @@ -30,7 +30,7 @@ trtis_model_overwrite=${14:-"False"}
3030
3131if [ " $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
3434else
3535 export BERT_DIR=data/download/google_pretrained_weights/uncased_L-12_H-768_A-12
3636fi
You can’t perform that action at this time.
0 commit comments