Update script.py
Browse files
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)
|