Commit 4584da0f authored by Franck Dary's avatar Franck Dary
Browse files

Removed warning

parent 8c6cc68d
......@@ -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;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment