From 69795e025d87d8428a02c195be01830f2671070e Mon Sep 17 00:00:00 2001 From: danielMaar <70383336+D-Maar@users.noreply.github.com> Date: Tue, 22 Oct 2024 16:32:27 +0200 Subject: [PATCH] Update model.R --- R/model.R | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/R/model.R b/R/model.R index 280d95d..4f7bcf7 100644 --- 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 } )