mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 11:24:00 +08:00
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>
This commit is contained in:
1 parent
6486c00d4d
commit
d9ee5752d2
1 file changed
+1
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user