We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8796f2e commit d935e2eCopy full SHA for d935e2e
im2latex/evaluation/evaluate_im2latex_inference.py
@@ -23,11 +23,11 @@ def test_evaluate(self):
23
dataset.prepare_data()
24
dataset.setup()
25
26
- inference = Im2LatexInference()
+ reasoner = Im2LatexInference()
27
trainer = pl.Trainer(gpus=1)
28
29
start_time = time.time()
30
- metrics = trainer.test(inference.lit_model, datamodule=dataset)
+ metrics = trainer.test(reasoner.lit_model, datamodule=dataset)
31
end_time = time.time()
32
33
test_bleu = round(metrics[0]["test_bleu"], 4)
0 commit comments