gigant commited on
Commit
18b0fa3
1 Parent(s): 9a6a1ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -24,6 +24,7 @@ def compute_batch_hashes(vid_path):
24
  h_prev = None
25
  batch = []
26
  for i in range(0, len(vr), DOWNSAMPLE * BATCH_SIZE):
 
27
  ids = [id for id in range(i, min(i + DOWNSAMPLE * BATCH_SIZE, len(vr)), DOWNSAMPLE)]
28
  vr.seek(0)
29
  batch = jnp.array(vr.get_batch(ids).asnumpy())
 
24
  h_prev = None
25
  batch = []
26
  for i in range(0, len(vr), DOWNSAMPLE * BATCH_SIZE):
27
+ print(f"batch_{i}")
28
  ids = [id for id in range(i, min(i + DOWNSAMPLE * BATCH_SIZE, len(vr)), DOWNSAMPLE)]
29
  vr.seek(0)
30
  batch = jnp.array(vr.get_batch(ids).asnumpy())