Those are really great ideas @Vaibhavbrkn @valhalla (and thanks for the datasets)!
@raghav66 Anybody can join!
Also I noticed lucidrains has a pretty awesome repo that should help so we can pull some items from it as well: DALLE-pytorch 3