Modify nn::AnyModule embedded in nn::Sequential

C++
1

Hi,

I have a custom module (let’s call it MyModule) that has conditional logic in its forward method. This custom module is part of an nn::Sequential.

I want to be able to do something along the following lines:


auto flag = true;
auto module = MyModule(flag);
auto sequential = nn::Sequential(
    module
);

auto x = torch::rand({T, inChannels});
auto y = sequential->forward(x);


sequential->modules()[0]->as<MyModule>->flag = false;

x = torch::rand({T, inChannels});
y = sequential->forward(x);

How do I go about this? The line

sequential->modules()[0]->as<MyModule>->flag = false;

doesn’t really set the flag (I think it only returns a temporary rvalue and changes it’s flag, but not of the underlying module).

Thanks.

2

Thanks, I am able to modify the underlying MyModule and my comment about rvalue is incorrect. I’m closing this issue.