python3 vqgan_eval.py --inference_type 'video' --patch_embed 'linear' --patch_size 8 --temporal_patch_size 4 --spatial_depth 4 --temporal_depth 4 --embedding_dim 512 --disc_layers 3 \
                      --enc_block "ttww" --dec_block "tttt" --twod_window_size 8 \
                      --causal_in_temporal_transformer --causal_in_peg --dim_head 64 --heads 8 --apply_noise --apply_blur --spatial_pos "rope" \
                      --n_codes 8192 --codebook_dim 8 --l2_code --commitment_weight 1.0 --no_random_restart \
                      --vqgan_ckpt "./ckpts_pub/imagenet_k600.ckpt" \
                      --batch_size 8 --data_path {PATH_TO_DATA_DIR} --train_datalist './annotations/k600_train.txt' --val_datalist './annotations/k600_val.txt' \
                      --loader_type "joint" --dataset 'imagenet_k600' --save ./k600 \
                      --resolution 256 --sequence_length 17 --discriminator_iter_start 0 --norm_type batch --replacewithgt 0 \
                      --perceptual_weight 4 --image_gan_weight 1 --video_gan_weight 1  --gan_feat_weight 4


python3 vqgan_eval.py --inference_type 'video' --patch_embed 'linear' --patch_size 8 --temporal_patch_size 4 --spatial_depth 4 --temporal_depth 4 --embedding_dim 512 --disc_layers 3 \
                      --enc_block "ttww" --dec_block "tttt" --twod_window_size 8 \
                      --causal_in_temporal_transformer --causal_in_peg --dim_head 64 --heads 8 --apply_noise --apply_blur --spatial_pos "rope" \
                      --n_codes 8192 --codebook_dim 8 --l2_code --commitment_weight 1.0 --no_random_restart \
                      --vqgan_ckpt "./ckpts_pub/imagenet_ucf.ckpt" \
                      --batch_size 8 --data_path {PATH_TO_DATA_DIR} --train_datalist './annotations/ucf_train.txt' --val_datalist './annotations/ucf_val.txt' \
                      --loader_type "joint" --dataset 'imagenet_ucf' --save ./ucf \
                      --resolution 256 --sequence_length 17 --discriminator_iter_start 0 --norm_type batch --replacewithgt 0 \
                      --perceptual_weight 4 --image_gan_weight 1 --video_gan_weight 1  --gan_feat_weight 4

python3 vqgan_eval.py --inference_type 'video' --patch_embed 'linear' --patch_size 8 --temporal_patch_size 4 --spatial_depth 4 --temporal_depth 4 --embedding_dim 512 --disc_layers 3 \
                      --enc_block "ttww" --dec_block "tttt" --twod_window_size 8 \
                      --causal_in_temporal_transformer --causal_in_peg --dim_head 64 --heads 8 --apply_noise --apply_blur --spatial_pos "rope" --use_vae \
                      --n_codes 8192 --codebook_dim 8 --l2_code --commitment_weight 1.0 --no_random_restart \
                      --vqgan_ckpt "./ckpts_pub/imagenet_ucf_vae.ckpt" \
                      --batch_size 8 --data_path {PATH_TO_DATA_DIR} --train_datalist './annotations/ucf_train.txt' --val_datalist './annotations/ucf_val.txt' \
                      --loader_type "joint" --dataset 'imagenet_ucf_vae' --save ./ucf --save_videos \
                      --resolution 256 --sequence_length 17 --discriminator_iter_start 0 --norm_type batch --replacewithgt 0 \
                      --perceptual_weight 4 --image_gan_weight 1 --video_gan_weight 1  --gan_feat_weight 4


python3 vqgan_eval.py --inference_type 'video' --patch_embed 'linear' --patch_size 8 --temporal_patch_size 4 --spatial_depth 4 --temporal_depth 4 --embedding_dim 512 --disc_layers 3 \
                      --enc_block "ttww" --dec_block "tttt" --twod_window_size 8 \
                      --causal_in_temporal_transformer --causal_in_peg --dim_head 64 --heads 8 --apply_noise --apply_blur --spatial_pos "rope" --use_vae \
                      --n_codes 8192 --codebook_dim 8 --l2_code --commitment_weight 1.0 --no_random_restart \
                      --vqgan_ckpt "./ckpts_pub/imagenet_k600_vae.ckpt" \
                      --batch_size 8 --data_path {PATH_TO_DATA_DIR} --train_datalist './annotations/k600_train.txt' --val_datalist './annotations/k600_val.txt' \
                      --loader_type "joint" --dataset 'imagenet_k600_vae' --save ./k600 --save_videos \
                      --resolution 256 --sequence_length 17 --discriminator_iter_start 0 --norm_type batch --replacewithgt 0 \
                      --perceptual_weight 4 --image_gan_weight 1 --video_gan_weight 1  --gan_feat_weight 4
