input = torch.randn(1, 3, 224, 224) macs, params = profile(model, inputs=(input, )) # 官方詳細範例import torch from torchvision import models
確定! 回上一頁