export DETECTRON2_DATASETS=/data/datasets

# good results for BYOL
python3 train_object_detection.py --config-file configs/pascal_voc_R_50_C4_24k_moco.yaml \
    --num-gpus 2 MODEL.WEIGHTS ./detectron_model.pkl SOLVER.IMS_PER_BATCH 16 SOLVER.BASE_LR 0.1


python3 convert_model_to_detectron2.py --pretrained_feature_extractor '/extra_room/junlin/SSL/solo-learn/trained_models/byol/r7zevbk4/byol-imagenet_so_1e-06_resnet50-r7zevbk4-ep=99.ckpt' 
--output_detectron_model ./detectron_model_byol_so_imagenet.pkl
