upload files

2023-02-13 17:20:19 +08:00
parent ea90485218
commit 1127f111d3
7 changed files with 756 additions and 410 deletions
@@ -50,7 +50,10 @@ def slice_segments(x, ids_str, segment_size=4):
  for i in range(x.size(0)):
    idx_str = ids_str[i]
    idx_end = idx_str + segment_size
-    ret[i] = x[i, :, idx_str:idx_end]
+    try:
+      ret[i] = x[i, :, idx_str:idx_end]
+    except RuntimeError:
+      print("?")
  return ret