liqian 1 year ago
parent
commit
4bd28afadf
4 changed files with 9 additions and 3 deletions
  1. 2 0
      ai_tag_task.sh
  2. 4 2
      asr_task.sh
  3. 1 1
      download_videos_task.py
  4. 2 0
      download_videos_task.sh

+ 2 - 0
ai_tag_task.sh

@@ -1,3 +1,5 @@
+ps -ef | grep ai_tag_task.py | grep -v grep | awk '{print $2}' | xargs kill -9
+
 cd /data/aigc-test
 
 source activate whisper

+ 4 - 2
asr_task.sh

@@ -1,8 +1,10 @@
+ps -ef | grep asr_task.py | grep -v grep | awk '{print $2}' | xargs kill -9
+
 cd /data/aigc-test
 
 source activate whisper
 
-CUDA_VISIBLE_DEVICES=0 python asr_task.py 0 > /data/aigc-test/logs/asr_task_0.log 2>&1 &
-CUDA_VISIBLE_DEVICES=1 python asr_task.py 1 > /data/aigc-test/logs/asr_task_1.log 2>&1
+CUDA_VISIBLE_DEVICES=0 python asr_task.py 0 > /data/aigc-test/logs/asr_task_0.log &
+CUDA_VISIBLE_DEVICES=1 python asr_task.py 1 > /data/aigc-test/logs/asr_task_1.log
 
 conda deactivate

+ 1 - 1
download_videos_task.py

@@ -59,7 +59,7 @@ def download_videos(project, table, dt):
     download_folder = 'videos'
     video_id_list = feature_df['videoid'].to_list()
     pool = multiprocessing.Pool(processes=6)
-    for video_id in video_id_list[:10]:
+    for video_id in video_id_list:
         video_path = feature_df[feature_df['videoid'] == video_id]['video_path'].values[0].strip()
         video_path = video_path.replace(' ', '')
         print(video_id, video_path)

+ 2 - 0
download_videos_task.sh

@@ -1,3 +1,5 @@
+ps -ef | grep download_videos_task.py | grep -v grep | awk '{print $2}' | xargs kill -9
+
 cd /data/aigc-test
 
 source activate whisper