train_dalle.py --vqgan_model_path imagenet_16384_slim.ckpt --vqgan_config_path imagenet_16384.yaml --taming --image_text_folder LAION/laion400m-dat-release/ --wds jpg,txt --shift_tokens --rotary_emb --random_resize_crop_lower_ratio 0.6 --epochs 20 --batch_size 3 --learning_rate 3e-4 --clip_grad_norm 0.5 --save_every_n_steps 1000 --keep_n_checkpoints 2 --dim 512 --depth 16 --heads 16 --text_seq_len 128 --truncate_captions --dim_head 64 --attn_dropout 0.1 --ff_dropout 0.1