bagua>=0.6,<0.7
torch
torchvision
