Spaces:
Running
on
T4
Running
on
T4
Commit
·
e059e75
1
Parent(s):
426dd72
Update vtoonify_model.py
Browse files- vtoonify_model.py +1 -1
vtoonify_model.py
CHANGED
@@ -208,7 +208,7 @@ class Model():
|
|
208 |
|
209 |
batch_frames = []
|
210 |
if video_cap.get(3) != 0:
|
211 |
-
batch_size = min(max(1, int(round(4 * 400 * 360 / video_cap.get(3) / video_cap.get(4)))),
|
212 |
else:
|
213 |
batch_size = 4
|
214 |
print('Using batch size of %d on %d frames'%(batch_size, num))
|
|
|
208 |
|
209 |
batch_frames = []
|
210 |
if video_cap.get(3) != 0:
|
211 |
+
batch_size = min(max(1, int(round(4 * 400 * 360 / video_cap.get(3) / video_cap.get(4)))), 4)
|
212 |
else:
|
213 |
batch_size = 4
|
214 |
print('Using batch size of %d on %d frames'%(batch_size, num))
|