Issue about the deployment on Spot

[weiqianwang123]

Hello, thank you very much for your open-source code. However, I encountered a problem when deploying the code on Spot. After initialization, once it enters explore mode, the code returns an action like tensor([[1]], device='cuda:0'), which leads to the following error:"linear": action[0][1].item(), IndexError: index 1 is out of bounds for dimension 0 with size 1
Could you please advise on how to resolve this issue?