File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff 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
2930python3 $VOLUME_SRCDIR /gen_qa_identity_models.py --tensorrt-shape-io --models_dir=$VOLUME_SHAPEDESTDIR
3031python3 $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
5556python3 $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
You can’t perform that action at this time.
0 commit comments