Skip to content

Commit ffa90e0

Browse files
committed
fix: the exception not using default topk argument
1 parent 715519a commit ffa90e0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

inference.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -119,8 +119,8 @@ def main():
119119
with open(os.path.join(args.output_dir, './topk_ids.csv'), 'w') as out_file:
120120
filenames = loader.dataset.filenames(basename=True)
121121
for filename, label in zip(filenames, topk_ids):
122-
out_file.write('{0},{1},{2},{3},{4},{5}\n'.format(
123-
filename, label[0], label[1], label[2], label[3], label[4]))
122+
out_file.write('{0},{1}\n'.format(
123+
filename, ','.join([ str(v) for v in label])))
124124

125125

126126
if __name__ == '__main__':

0 commit comments

Comments
 (0)