Display results for all children of nn.Sequential

pfnet-research / chainer-compiler

Experimental toolchain to compile and run Chainer models

MIT License

112 stars 23 forks source link

Display results for all children of nn.Sequential #819

Closed momohatt closed 4 years ago

momohatt commented 4 years ago

By using nn.Sequential, invoking __call__ method of an nn.Model object in PyTorch can call more than two forward functions. However, current implementation of inference engine does not consider such cases. This PR fixes this behavior by changing the type of subroutine_node attribute of InferenceEngine from Dict[gast.Call, gast.FunctionDef] to Dict[gast.Call, List[gast.FunctionDef]].

shinh commented 4 years ago

/test

pfn-ci-bot commented 4 years ago

Successfully created a job for commit fae40a8:

Dashboard for commit fae40a8