Ahmet Iscen
Results
1
comments of
Ahmet Iscen
Hi, 1. log operation is already applied to logits on Line 44: s_softmax_temp = jax.nn.log_softmax(logits / temperature) # [n, d] No need to do it again. 2. This is a...