Uky

Hi! Thanks for the excellent work. I am trying to do inference with the checkpoint provided. I am using the command "main.py --model init --dataset htm-370k --batch_size 128 --use_text_pos_enc 0...