An error occurred while fetching folder content.
Guy Jacob
authored
* By duplicate modules we mean: self.relu1 = nn.Relu() self.relu2 = self.relu1 * The issue: The second module ('relu2') will not be returned by torch.nn.Module.named_modules/children() * When converting to DistillerModuleList, in order to maintain the original order of modules and in order to have a correct mapping of names before/after the conversion - we need to take the duplicates into account * Implemented an internal version of named_modules/children that includes duplicates * Added test case for this + refactored the module list conversion tests
Name | Last commit | Last update |
---|