import torch
from torchsummary import summary
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = torch.load("torch-save.pkl")
summary(model.to(device), input_size=(3, 208, 976), batch_size=1)
07-31
554
11-18
2032
07-03
1263