├── .gitignore ├── LICENSE ├── README.md ├── config ├── base.py └── dgx.py ├── ddpo_pytorch ├── aesthetic_scorer.py ├── assets │ ├── activities.txt │ ├── activities_v0.txt │ ├── imagenet_classes.txt │ ├── sac+logos+ava1-l14-linearMSE.pth │ └── simple_animals.txt ├── diffusers_patch │ ├── ddim_with_logprob.py │ └── pipeline_with_logprob.py ├── prompts.py ├── rewards.py └── stat_tracking.py ├── scripts └── train.py ├── setup.py └── teaser.jpg /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /config/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/config/base.py -------------------------------------------------------------------------------- /config/dgx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/config/dgx.py -------------------------------------------------------------------------------- /ddpo_pytorch/aesthetic_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/aesthetic_scorer.py -------------------------------------------------------------------------------- /ddpo_pytorch/assets/activities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/assets/activities.txt -------------------------------------------------------------------------------- /ddpo_pytorch/assets/activities_v0.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/assets/activities_v0.txt -------------------------------------------------------------------------------- /ddpo_pytorch/assets/imagenet_classes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/assets/imagenet_classes.txt -------------------------------------------------------------------------------- /ddpo_pytorch/assets/sac+logos+ava1-l14-linearMSE.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/assets/sac+logos+ava1-l14-linearMSE.pth -------------------------------------------------------------------------------- /ddpo_pytorch/assets/simple_animals.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/assets/simple_animals.txt -------------------------------------------------------------------------------- /ddpo_pytorch/diffusers_patch/ddim_with_logprob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/diffusers_patch/ddim_with_logprob.py -------------------------------------------------------------------------------- /ddpo_pytorch/diffusers_patch/pipeline_with_logprob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/diffusers_patch/pipeline_with_logprob.py -------------------------------------------------------------------------------- /ddpo_pytorch/prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/prompts.py -------------------------------------------------------------------------------- /ddpo_pytorch/rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/rewards.py -------------------------------------------------------------------------------- /ddpo_pytorch/stat_tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/ddpo_pytorch/stat_tracking.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/scripts/train.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/setup.py -------------------------------------------------------------------------------- /teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kvablack/ddpo-pytorch/HEAD/teaser.jpg --------------------------------------------------------------------------------