Skip to content

GitLab

Commit 6c2f36de authored by Yang Yong(雍洋)'s avatar Yang Yong(雍洋) Committed by GitHub
Browse files

update profile (#273)

parent 89ce2aa6
Show whitespace changes
Inline Side-by-side
Showing with 9 additions and 8 deletions
lightx2v/models/runners/default_runner.py
View file @ 6c2f36de
...@@ -217,6 +217,7 @@ class DefaultRunner(BaseRunner): ...@@ -217,6 +217,7 @@ class DefaultRunner(BaseRunner):
def run_main(self, total_steps=None): def run_main(self, total_steps=None):
self.init_run() self.init_run()
for segment_idx in range(self.video_segment_num): for segment_idx in range(self.video_segment_num):
with ProfilingContext4Debug(f"segment end2end {segment_idx}"):
# 1. default do nothing # 1. default do nothing
self.init_run_segment(segment_idx) self.init_run_segment(segment_idx)
# 2. main inference loop # 2. main inference loop
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment