David Novotny 9f14e82b5a SO3 improvements for stable gradients.
Summary:
Improves so3 functions to make gradient computation stable:
- Instead of `torch.acos`, uses `acos_linear_extrapolation` which has finite gradients of reasonable magnitude for all inputs.
- Adds tests for the latter.

The tests of the finiteness of the gradient in `test_so3_exp_singularity`, `test_so3_exp_singularity`, `test_so3_cos_bound` would fail if the `so3` functions would call `torch.acos` instead of `acos_linear_extrapolation`.

Reviewed By: bottler

Differential Revision: D23326429

fbshipit-source-id: dc296abf2ae3ddfb3942c8146621491a9cb740ee
2021-06-21 04:48:27 -07:00
..
2021-06-11 14:33:40 -07:00
2020-01-23 11:53:46 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2021-01-05 03:59:20 -08:00
2021-01-20 13:08:35 -08:00
2021-01-06 06:23:48 -08:00
2021-01-06 06:23:48 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2020-11-03 13:06:35 -08:00
2021-05-26 04:54:19 -07:00
2020-05-04 09:56:44 -07:00
2020-04-24 09:11:04 -07:00
2020-05-05 11:09:45 -07:00
2020-04-24 09:11:04 -07:00
2020-08-07 01:59:58 -07:00
2021-05-26 04:54:19 -07:00
2021-05-04 05:36:51 -07:00
2020-07-29 05:01:07 -07:00
2020-11-30 11:38:04 -08:00
2020-04-06 06:41:00 -07:00
2020-04-24 09:11:04 -07:00
2021-04-09 09:58:54 -07:00
2021-04-09 09:58:54 -07:00
2021-04-12 19:10:18 -07:00
2021-04-08 09:38:43 -07:00
2021-04-09 09:58:54 -07:00
2021-04-12 19:10:18 -07:00
2021-04-09 09:58:54 -07:00
2021-01-06 04:01:29 -08:00
2021-04-12 19:10:18 -07:00
2021-04-09 09:58:54 -07:00
2021-06-11 13:38:46 -07:00
2020-01-23 11:53:46 -08:00