bagua>=0.6
torch
torchvision
