BASE-Pytorch