Update model.R

D-Maar · Oct 22, 2024 · 69795e0 · 69795e0
1 parent f5ef138
commit 69795e0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/R/model.R b/R/model.R
@@ -208,7 +208,7 @@ inceptionBlock_A_2D = torch::nn_module(
     # print(branchBRes$size())
     # print(branchCRes$size())
     # print(branchDRes$size())
-    res = torch_cat(list(branchARes, branchBRes, branchCRes, branchDRes),2L)
+    res = torch::torch_cat(list(branchARes, branchBRes, branchCRes, branchDRes),2L)
     res
   }
 )
@@ -298,7 +298,7 @@ inceptionBlock_A_2D_reduction = torch::nn_module(
     # print(branchBRes$size())
     # print(branchCRes$size())
     # print(branchDRes$size())
-    res = torch_cat(list(branchARes, branchBRes, branchCRes, branchDRes),2L)
+    res = torch::torch_cat(list(branchARes, branchBRes, branchCRes, branchDRes),2L)
     res
   }
 )
@@ -392,7 +392,7 @@ inceptionBlock_A_1D = torch::nn_module(
     # print(branchBRes$size())
     # print(branchCRes$size())
     # print(branchDRes$size())
-    res = torch_cat(list(branchARes, branchBRes, branchCRes, branchDRes),2L)
+    res = torch::torch_cat(list(branchARes, branchBRes, branchCRes, branchDRes),2L)
     res
   }
 )