smilehan
Results
1
comments of
smilehan
I meet the question so. It cause bytorch version. In latest torch, 0-dim tensor is invaild, 我也遇到这问题，是torch版本引起的，稍作修改即可。 Change code line77 ` action = torch.argmax(prediction)[0] ` To ` action = torch.argmax(prediction)...