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__':
|