torch
torchvision
openmim
gradio
einops
tensorboardX
easydict