This commit is contained in:
James Betker 2021-08-30 21:45:58 -06:00
parent f1a0c21fb2
commit 274d352e6f

View File

@ -75,7 +75,7 @@ if __name__ == "__main__":
continue
pred = forward_pass(model, data, dataset_dir, opt, batch)
pred = pred.replace('_', '')
output.write(f'{pred}\t{os.path.basename(data["path"][0])}')
output.write(f'{pred}\t{os.path.basename(data["path"][0])}\n')
output.flush()
batch += 1