Pytorch Module

If the memeber module is typed with optional, it does not registered as a module and remains none even if we assigned it. So make sure to declare it as a non-optional member.