torch
torchvision
ptflops
