Searched refs:ModelForConvTransposeBNFusion (Results 1 – 3 of 3) sorted by relevance
30 ModelForConvTransposeBNFusion,351 model = ModelForConvTransposeBNFusion().train()
2437 class ModelForConvTransposeBNFusion(nn.Module): class
173 ModelForConvTransposeBNFusion,569 m = ModelForConvTransposeBNFusion().eval()