axel588

@HReynaud What training time do you have for training on 10 000 images on text to image ?

@lucidrains I applied the attention layer at first, but even with a dimension of 8 ( very low yes ) and a batch of 1 it overflows my 24gb memory...