├── .gitignore ├── README.md ├── assets └── comparison.png ├── requirements.txt ├── scripts ├── train.sh └── validate_checkpoint.sh └── src ├── data_set.py ├── losses.py ├── train_pair_pos_neg.py └── validate.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Picsart-AI-Research/Social-Reward/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Picsart-AI-Research/Social-Reward/HEAD/README.md -------------------------------------------------------------------------------- /assets/comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Picsart-AI-Research/Social-Reward/HEAD/assets/comparison.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Picsart-AI-Research/Social-Reward/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Picsart-AI-Research/Social-Reward/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /scripts/validate_checkpoint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Picsart-AI-Research/Social-Reward/HEAD/scripts/validate_checkpoint.sh -------------------------------------------------------------------------------- /src/data_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Picsart-AI-Research/Social-Reward/HEAD/src/data_set.py -------------------------------------------------------------------------------- /src/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Picsart-AI-Research/Social-Reward/HEAD/src/losses.py -------------------------------------------------------------------------------- /src/train_pair_pos_neg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Picsart-AI-Research/Social-Reward/HEAD/src/train_pair_pos_neg.py -------------------------------------------------------------------------------- /src/validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Picsart-AI-Research/Social-Reward/HEAD/src/validate.py --------------------------------------------------------------------------------