Include camera geometry in lightning checkpoints
Would be very useful in production mode.
Possible hack: define geometry as a hyperparameter and use lightning save_hyperparameter method in module
Would be very useful in production mode.
Possible hack: define geometry as a hyperparameter and use lightning save_hyperparameter method in module