diff --git a/inference_mp4_2x.py b/inference_mp4_2x.py index edcb557..f866515 100644 --- a/inference_mp4_2x.py +++ b/inference_mp4_2x.py @@ -60,7 +60,7 @@ while success: p = (F.interpolate(I0, (16, 16), mode='bilinear', align_corners=False) - F.interpolate(I1, (16, 16), mode='bilinear', align_corners=False)).abs().mean() if p < 1e-3 and args.skip: - if cnt % 10 == 0: + if cnt % 100 == 0: print("Warning: Your video has {} static frames, skipping them change the duration of the generated video.".format(cnt)) cnt += 1 pbar.update(1) diff --git a/inference_mp4_4x.py b/inference_mp4_4x.py index 29e54e3..27d90b7 100644 --- a/inference_mp4_4x.py +++ b/inference_mp4_4x.py @@ -60,7 +60,7 @@ while success: p = (F.interpolate(I0, (16, 16), mode='bilinear', align_corners=False) - F.interpolate(I1, (16, 16), mode='bilinear', align_corners=False)).abs().mean() if p < 1e-3 and args.skip: - if cnt % 10 == 0: + if cnt % 100 == 0: print("Warning: Your video has {} static frames, skipping them may change the duration of the generated video.".format(cnt)) cnt += 1 pbar.update(1)