We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1b366bc commit f6d43acCopy full SHA for f6d43ac
1 file changed
nerfstudio/field_components/activations.py
@@ -21,8 +21,8 @@
21
import torch
22
from jaxtyping import Float
23
from torch import Tensor
24
-from torch.autograd import Function
25
from torch.amp import custom_bwd, custom_fwd
+from torch.autograd import Function
26
27
28
class _TruncExp(Function):
0 commit comments