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:
Ethan WeberandMatias Turkulainen authored and GitHub committed 2023-06-27 09:33:28 -07:00
1 parent 6486c00d4d
commit d9ee5752d2
1 file changed
+1
+1
View File
@@ -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)