CUDA_VISIBLE_DEVICED=0 python3 main.py --kge_n_epoch 100 --n_epoch 1
- [--n_epoch N_EPOCH]
- [--drop_out DROP_OUT]
- [--batch_size BATCH_SIZE]
- [--lr LR]
- [--use_kge USE_KGE]
- [--use_kge_pt USE_KGE_PT]
- [--image_model {vit,resnet}]
- [--device DEVICE]
- [--ans1_only ANS1_ONLY]
- [--ans2_only ANS2_ONLY]
- [--both BOTH]
- [--max_token MAX_TOKEN]
- [--kge_n_epoch KGE_N_EPOCH]
- [--kge_lr KGE_LR]
- [--kge_batch KGE_BATCH]
- [--kge_margin KGE_MARGIN]
- [--kge_conv_size KGE_CONV_SIZE]