From d9ee5752d20212a867da0c6cf67ba6558206b132 Mon Sep 17 00:00:00 2001 From: Ethan Weber Date: Tue, 27 Jun 2023 09:33:28 -0700 Subject: [PATCH] Fix to fall back to torch if tcnn not found (#2124) fix for falling back to torch if tcnn not found Co-authored-by: Matias Turkulainen <30566358+maturk@users.noreply.github.com> --- nerfstudio/field_components/mlp.py | 1 + 1 file changed, 1 insertion(+) diff --git a/nerfstudio/field_components/mlp.py b/nerfstudio/field_components/mlp.py index 2d766e233..52eabd29d 100644 --- a/nerfstudio/field_components/mlp.py +++ b/nerfstudio/field_components/mlp.py @@ -102,6 +102,7 @@ class MLP(FieldComponent): self.build_nn_modules() elif implementation == "tcnn" and not TCNN_EXISTS: print_tcnn_speed_warning("MLP") + self.build_nn_modules() elif implementation == "tcnn": activation_str = activation_to_tcnn_string(activation) output_activation_str = activation_to_tcnn_string(out_activation)