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")
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")