From 4584da0f6002b68a37ac7d1da4108a10d691b050 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Sat, 20 Mar 2021 20:38:58 +0100
Subject: [PATCH] Removed warning
---
torch_modules/src/ModularNetwork.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/torch_modules/src/ModularNetwork.cpp b/torch_modules/src/ModularNetwork.cpp
index 066b79d..89b2de0 100644
--- a/torch_modules/src/ModularNetwork.cpp
+++ b/torch_modules/src/ModularNetwork.cpp
@@ -99,7 +99,7 @@ torch::Tensor ModularNetworkImpl::forward(torch::Tensor input, const std::string
torch::Tensor ModularNetworkImpl::extractContext(Config & config)
{
- torch::Tensor context = torch::zeros({totalInputSize}, torch::TensorOptions(torch::kLong).device(NeuralNetworkImpl::getDevice()));
+ torch::Tensor context = torch::zeros({(long)totalInputSize}, torch::TensorOptions(torch::kLong).device(NeuralNetworkImpl::getDevice()));
for (auto & mod : modules)
mod->addToContext(context, config);
return context;
--
GitLab