why set Rigid.identity trainable?

[KK666-AI]

I notice that Rigid.identity here requires gradient in training stage. It will be inconsistent with the stage of inference? Is it correct?

[gahdritz]

During inference, whether the rigid object has grad enabled doesn't matter since grad computation is disabled globally.

[KK666-AI]

is rigid.identity trainable during training phage？