A suggestion about README.
lixiang007666 opened this issue · comments
input = torch.randn(1, 3, 224, 224)replace with
input = torch.randn(1, 3, 224, 224) #.to("cuda")Lyken17 / pytorch-OpCounter
Count the MACs / FLOPs of your PyTorch model.
Count the MACs / FLOPs of your PyTorch model.
lixiang007666 opened this issue · comments
input = torch.randn(1, 3, 224, 224)replace with
input = torch.randn(1, 3, 224, 224) #.to("cuda")