ShanghaiTech Datamodule

ShanghaiTech Datamodule#