Spaces:
Running
Running
fix csv output
Browse files- predict.py +3 -2
predict.py
CHANGED
|
@@ -77,8 +77,9 @@ def main():
|
|
| 77 |
pass
|
| 78 |
for batch in tqdm.tqdm(loader):
|
| 79 |
scores = scorer.score(batch.to(device))
|
| 80 |
-
with open(args.out_path, 'a') as fw:
|
| 81 |
-
|
|
|
|
| 82 |
|
| 83 |
|
| 84 |
if __name__ == '__main__':
|
|
|
|
| 77 |
pass
|
| 78 |
for batch in tqdm.tqdm(loader):
|
| 79 |
scores = scorer.score(batch.to(device))
|
| 80 |
+
with open(args.out_path, 'a') as fw:
|
| 81 |
+
for s in scores:
|
| 82 |
+
fw.write(str(s) + "\n")
|
| 83 |
|
| 84 |
|
| 85 |
if __name__ == '__main__':
|