model = EfficientNet.from_pretrained('efficientnet-b0') device = torch.device(“cuda”) model.to(device) dummy_input = torch.randn(1, 3, 224, ...
確定! 回上一頁