Closed scene-the-ella closed 1 month ago
In my reproduction, that was not the problem. There was a key "results" in the output dictionary.
In my reproduction, that was not the problem. There was a key "results" in the output dictionary.
Hi @jisoo0-0 ! Thank you for your appreciation of my issue in responding on behalf of the author! However, It seems that what you're saying now is exactly the same as my initial question. I'm actually arguing that output['results'][-1]
is correct, not output[self.args.output][-1]
. 😉
Hi, you can change it to the key you want.
Hi! Thank you for your nice work! BTW, isn't it
pred = output['results'][-1].detach()
instead ofpred = output[self.args.output][-1].detach()
? Cuz that line causes KeyError.