From 7214f8ddbc53e181bf71378739462207765a67f6 Mon Sep 17 00:00:00 2001 From: huangjun12 Date: Mon, 20 Mar 2023 11:40:52 +0000 Subject: [PATCH 1/2] add compile for tsm --- .../scripts/TSM/benchmark_common/run_benchmark.sh | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/frame_benchmark/pytorch/dynamic/PaddleVideo/scripts/TSM/benchmark_common/run_benchmark.sh b/frame_benchmark/pytorch/dynamic/PaddleVideo/scripts/TSM/benchmark_common/run_benchmark.sh index c204fbf45a..8b548056e1 100755 --- a/frame_benchmark/pytorch/dynamic/PaddleVideo/scripts/TSM/benchmark_common/run_benchmark.sh +++ b/frame_benchmark/pytorch/dynamic/PaddleVideo/scripts/TSM/benchmark_common/run_benchmark.sh @@ -54,7 +54,11 @@ function _train(){ --epochs ${max_epoch} \ --batch-size ${batch_size} \ -j ${num_workers}" - + + if [ "${FLAG_TORCH_COMPILE}" = "True" ] || [ "${FLAG_TORCH_COMPILE}" = "true" ];then + train_options="${train_options} --torchcompile" + fi + nodes="${device_num:1:1}" if [[ nodes -gt 1 ]];then From c5918bd55ed03050c398a0a907d503158b7b8cd2 Mon Sep 17 00:00:00 2001 From: huangjun12 <2399845970@qq.com> Date: Tue, 21 Mar 2023 02:41:20 +0000 Subject: [PATCH 2/2] update submodule --- .../pytorch/dynamic/PaddleVideo/models/temporal-shift-module | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/frame_benchmark/pytorch/dynamic/PaddleVideo/models/temporal-shift-module b/frame_benchmark/pytorch/dynamic/PaddleVideo/models/temporal-shift-module index d199e7a07f..a69298e631 160000 --- a/frame_benchmark/pytorch/dynamic/PaddleVideo/models/temporal-shift-module +++ b/frame_benchmark/pytorch/dynamic/PaddleVideo/models/temporal-shift-module @@ -1 +1 @@ -Subproject commit d199e7a07f550acf1b12dba1954bb59ecdc001b3 +Subproject commit a69298e631ed8f1984323951dc5416a8e080fc47