You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I was running sac_continuous_action_torchcompile.py and wanted to do a comparison of compile vs no compile on Mac M3. I changed line 172 to use MPS instead of CUDA.
I ran into a weird issue on line 222 which calculates the MSE loss:
RuntimeError: impl: target and input tensors must have identical shapes
I checked val.view(-1) and next_q_value actually have the same shape. So I don't understand why this error happens.
I was able to solve it by changing to the following loss computation:
Hi, thanks for this cool project!
I was running
sac_continuous_action_torchcompile.py
and wanted to do a comparison of compile vs no compile on Mac M3. I changed line 172 to use MPS instead of CUDA.I ran into a weird issue on line 222 which calculates the MSE loss:
I checked
val.view(-1)
andnext_q_value
actually have the same shape. So I don't understand why this error happens.I was able to solve it by changing to the following loss computation:
I don't know if this will affect performance significantly (I doubt).
I installed both
torchrl
andtensordict
from source if that matters.The text was updated successfully, but these errors were encountered: