jeeyung

Hello, https://github.com/bbdamodaran/deepJDOT/blob/b8814679d0e4b6f7c3e5aacf3964b767d1d2c22d/Deepjdot.py#L200 Why didn't you use log (softmax output) when you get C1 in Deepjdot.py ?

Hi! I think the shape of logits from `self.tokens_to_logits` is [batch, 257, 1026] because you defined self.tok_emb = nn.Embedding(args.num_codebook_vectors + 2, args.dim). However, the number of codebook's embedding is 1024...

Hi, I used 32x32 size images. With 12GB GPUs, I reduce the batch size from 16 to 2 or 1(per gpu). However, still I got CUDA out of memory error....