gmancino-ball commited on
Commit
28d2ddc
·
verified ·
1 Parent(s): 55a1a9f

Update script.py

Browse files
Files changed (1) hide show
  1. script.py +0 -2
script.py CHANGED
@@ -89,7 +89,6 @@ model = Model().to(device)
89
 
90
  # iterate over the dataset
91
  out = []
92
- acc = []
93
  for el in tqdm.tqdm(dataset_remote):
94
 
95
  # start_time = time.time()
@@ -99,7 +98,6 @@ for el in tqdm.tqdm(dataset_remote):
99
  # el["video"]["path"] containts the filename. This is just for reference and you cant actually load it
100
 
101
  # if you are using libraries that expect a file. You can use BytesIO object
102
- acc.append(torch.randn((25000, 25000)).to("cuda:0"))
103
  try:
104
  file_like = io.BytesIO(el["video"]["bytes"])
105
  tensor = preprocess(file_like, device=device)
 
89
 
90
  # iterate over the dataset
91
  out = []
 
92
  for el in tqdm.tqdm(dataset_remote):
93
 
94
  # start_time = time.time()
 
98
  # el["video"]["path"] containts the filename. This is just for reference and you cant actually load it
99
 
100
  # if you are using libraries that expect a file. You can use BytesIO object
 
101
  try:
102
  file_like = io.BytesIO(el["video"]["bytes"])
103
  tensor = preprocess(file_like, device=device)