Plotting routines fail for segments that contain tensors with require_grad=True
#287
Labels
bug
Something isn't working
require_grad=True
#287
The plotting routines of Cheetah currently fail if the
Segment
they are called on contain tensors withrequire_grad=True
. This is because Matplotlib internally callstensor.numpy()
, which fails for tensors that include gradients.The text was updated successfully, but these errors were encountered: