Skip to content

Commit 5f24e68

Browse files
committed
Copy the model generation scripts to model build directory
1 parent eadd6f8 commit 5f24e68

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

qa/common/run_trt_model_generation.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ mkdir -p $VOLUME_BUILD_DIR $VOLUME_SRCDIR $VOLUME_DESTDIR $VOLUME_SHAPEDESTDIR \
2525
$VOLUME_VARDESTDIR $VOLUME_RESHAPEDESTDIR $VOLUME_DYNASEQDESTDIR \
2626
$VOLUME_DYNASEQIMPLICITDESTDIR
2727

28+
cp *.py $VOLUME_SRCDIR
2829
# Models using shape tensor i/o
2930
python3 $VOLUME_SRCDIR/gen_qa_identity_models.py --tensorrt-shape-io --models_dir=$VOLUME_SHAPEDESTDIR
3031
python3 $VOLUME_SRCDIR/gen_qa_sequence_models.py --tensorrt-shape-io --models_dir=$VOLUME_SHAPEDESTDIR
@@ -53,4 +54,4 @@ chmod -R 777 $VOLUME_DYNASEQDESTDIR
5354

5455
# Implicit sequence models
5556
python3 $VOLUME_SRCDIR/gen_qa_dyna_sequence_implicit_models.py --tensorrt --models_dir=$VOLUME_DYNASEQIMPLICITDESTDIR
56-
chmod -R 777 $VOLUME_DYNASEQIMPLICITDESTDIR
57+
chmod -R 777 $VOLUME_DYNASEQIMPLICITDESTDIR

0 commit comments

Comments
 (0)