Spaces:
Running
on
T4
Running
on
T4
PKUWilliamYang
commited on
Commit
•
eaba2a3
1
Parent(s):
d376f4c
Update vtoonify_model.py
Browse files- vtoonify_model.py +1 -1
vtoonify_model.py
CHANGED
@@ -242,7 +242,7 @@ class Model():
|
|
242 |
batch_size = min(max(1, int(4 * 400 * 360/ video_cap.get(3) / video_cap.get(4))), 4)
|
243 |
else:
|
244 |
batch_size = 1
|
245 |
-
print('*** Toonify using batch size of %d on %dx%d video of %d frames'%(batch_size, int(video_cap.get(3)*4), int(video_cap.get(4)*4, num))
|
246 |
with torch.no_grad():
|
247 |
if self.color_transfer:
|
248 |
s_w = exstyle
|
|
|
242 |
batch_size = min(max(1, int(4 * 400 * 360/ video_cap.get(3) / video_cap.get(4))), 4)
|
243 |
else:
|
244 |
batch_size = 1
|
245 |
+
print('*** Toonify using batch size of %d on %dx%d video of %d frames'%(batch_size, int(video_cap.get(3)*4), int(video_cap.get(4)*4), num))
|
246 |
with torch.no_grad():
|
247 |
if self.color_transfer:
|
248 |
s_w = exstyle
|