Searched refs:LinearModelWithSubmodule (Results 1 – 3 of 3) sorted by relevance
12 LinearModelWithSubmodule,79 model = LinearModelWithSubmodule().eval()
179 LinearModelWithSubmodule,3073 model = LinearModelWithSubmodule().train()3109 orig = LinearModelWithSubmodule().eval()
1608 class LinearModelWithSubmodule(nn.Module): class