cmd.txt

python train.py --video_data_root_list datasets/images/images \
                --flow_data_root_list datasets/flow_fp16/flow_fp16 \
                --mask_data_root_list datasets/pgm/pgm \
                --data_root_imagenet datasets/imgnet \
                --annotation_file_path datasets/final_annot.csv \
                --imagenet_pairs_file datasets/pairs.txt \
                --gpu_ids 0 \
                --workers 12 \
                --batch_size 2 \
                --real_reference_probability 0.99 \
                --weight_contextual 1 \
                --weight_perceptual 0.1 \
                --weight_smoothness 5 \
                --weight_gan 0.9 \
                --weight_consistent 0.1 \
                --use_wandb True \
                --wandb_token "f05d31e6b15339b1cfc5ee1c77fe51f66fc3ea9e" \
                --wandb_name "vit_tiny_patch16_384_nofeat" \
                --checkpoint_step 500 \
                --epoch_train_discriminator 3 \
                --epoch 20