Searched refs:CustomFuncStrideModule (Results 1 – 1 of 1) sorted by relevance
166 class CustomFuncStrideModule(torch.nn.Module): class280 model = CustomFuncStrideModule()