torch
torchvision
thop
torchinfo
