├── LICENSE ├── README.md ├── requirements.txt ├── setup.py └── trainer ├── accelerators ├── __init__.py ├── base_accelerator.py ├── debug_accelerator.py ├── deepspeed_accelerator.py └── utils.py ├── conf ├── config.yaml ├── experiment │ ├── clip.yaml │ └── clip_h.yaml ├── slurm │ └── stability.yaml └── slurm_config.yaml ├── configs ├── __init__.py └── configs.py ├── criterions ├── __init__.py └── clip_criterion.py ├── datasetss ├── __init__.py ├── base_dataset.py └── clip_hf_dataset.py ├── lr_schedulers ├── __init__.py ├── constant_with_warmup.py └── dummy_lr_scheduler.py ├── models ├── __init__.py ├── base_model.py └── clip_model.py ├── optimizers ├── __init__.py ├── adamw.py └── dummy_optimizer.py ├── scripts ├── eval_preference_predictor.py └── train.py ├── slurm_scripts └── slurm_train.py ├── tasks ├── __init__.py ├── base_task.py └── clip_task.py └── utils ├── FID ├── __init__.py ├── fid_score.py ├── img_data.py └── inception.py ├── __init__.py ├── data_utils.py └── slurm_utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/setup.py -------------------------------------------------------------------------------- /trainer/accelerators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/accelerators/__init__.py -------------------------------------------------------------------------------- /trainer/accelerators/base_accelerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/accelerators/base_accelerator.py -------------------------------------------------------------------------------- /trainer/accelerators/debug_accelerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/accelerators/debug_accelerator.py -------------------------------------------------------------------------------- /trainer/accelerators/deepspeed_accelerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/accelerators/deepspeed_accelerator.py -------------------------------------------------------------------------------- /trainer/accelerators/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/accelerators/utils.py -------------------------------------------------------------------------------- /trainer/conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/conf/config.yaml -------------------------------------------------------------------------------- /trainer/conf/experiment/clip.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/conf/experiment/clip.yaml -------------------------------------------------------------------------------- /trainer/conf/experiment/clip_h.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/conf/experiment/clip_h.yaml -------------------------------------------------------------------------------- /trainer/conf/slurm/stability.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/conf/slurm/stability.yaml -------------------------------------------------------------------------------- /trainer/conf/slurm_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/conf/slurm_config.yaml -------------------------------------------------------------------------------- /trainer/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/configs/__init__.py -------------------------------------------------------------------------------- /trainer/configs/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/configs/configs.py -------------------------------------------------------------------------------- /trainer/criterions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/criterions/__init__.py -------------------------------------------------------------------------------- /trainer/criterions/clip_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/criterions/clip_criterion.py -------------------------------------------------------------------------------- /trainer/datasetss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/datasetss/__init__.py -------------------------------------------------------------------------------- /trainer/datasetss/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/datasetss/base_dataset.py -------------------------------------------------------------------------------- /trainer/datasetss/clip_hf_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/datasetss/clip_hf_dataset.py -------------------------------------------------------------------------------- /trainer/lr_schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/lr_schedulers/__init__.py -------------------------------------------------------------------------------- /trainer/lr_schedulers/constant_with_warmup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/lr_schedulers/constant_with_warmup.py -------------------------------------------------------------------------------- /trainer/lr_schedulers/dummy_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/lr_schedulers/dummy_lr_scheduler.py -------------------------------------------------------------------------------- /trainer/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/models/__init__.py -------------------------------------------------------------------------------- /trainer/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/models/base_model.py -------------------------------------------------------------------------------- /trainer/models/clip_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/models/clip_model.py -------------------------------------------------------------------------------- /trainer/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/optimizers/__init__.py -------------------------------------------------------------------------------- /trainer/optimizers/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/optimizers/adamw.py -------------------------------------------------------------------------------- /trainer/optimizers/dummy_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/optimizers/dummy_optimizer.py -------------------------------------------------------------------------------- /trainer/scripts/eval_preference_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/scripts/eval_preference_predictor.py -------------------------------------------------------------------------------- /trainer/scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/scripts/train.py -------------------------------------------------------------------------------- /trainer/slurm_scripts/slurm_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/slurm_scripts/slurm_train.py -------------------------------------------------------------------------------- /trainer/tasks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/tasks/__init__.py -------------------------------------------------------------------------------- /trainer/tasks/base_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/tasks/base_task.py -------------------------------------------------------------------------------- /trainer/tasks/clip_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/tasks/clip_task.py -------------------------------------------------------------------------------- /trainer/utils/FID/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trainer/utils/FID/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/utils/FID/fid_score.py -------------------------------------------------------------------------------- /trainer/utils/FID/img_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/utils/FID/img_data.py -------------------------------------------------------------------------------- /trainer/utils/FID/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/utils/FID/inception.py -------------------------------------------------------------------------------- /trainer/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trainer/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/utils/data_utils.py -------------------------------------------------------------------------------- /trainer/utils/slurm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuvalkirstain/PickScore/HEAD/trainer/utils/slurm_utils.py --------------------------------------------------------------------------------