forked from mrq/DL-Art-School
dug
This commit is contained in:
parent
f1a0c21fb2
commit
274d352e6f
|
@ -75,7 +75,7 @@ if __name__ == "__main__":
|
||||||
continue
|
continue
|
||||||
pred = forward_pass(model, data, dataset_dir, opt, batch)
|
pred = forward_pass(model, data, dataset_dir, opt, batch)
|
||||||
pred = pred.replace('_', '')
|
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()
|
output.flush()
|
||||||
batch += 1
|
batch += 1
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user