search

Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.
MIT License
4.9k stars 528 forks source link

A suggestion about README. #192

Closed lixiang007666 closed 2 years ago

lixiang007666 commented 2 years ago 
input = torch.randn(1, 3, 224, 224)

replace with

input = torch.randn(1, 3, 224, 224) #.to("cuda")