1d04211666
* update from scratch configs * update gym pretraining configs - use fewer epochs * update robomimic pretraining configs - use fewer epochs * allow trajectory plotting in eval agent * add simple vit unet * update avoid pretraining configs - use fewer epochs * update furniture pretraining configs - use same amount of epochs as before * add robomimic diffusion unet pretraining configs * update robomimic finetuning configs - higher lr * add vit unet checkpoint urls * update pretraining and finetuning instructions as configs are updated
84 lines
1.6 KiB
TOML
84 lines
1.6 KiB
TOML
[build-system]
|
|
requires = ["setuptools>=61.0"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "dppo"
|
|
version = "0.7.0"
|
|
description = "Fine-tuning diffusion policies with PPO."
|
|
readme = "README.md"
|
|
requires-python = ">=3.8"
|
|
classifiers = [
|
|
"Programming Language :: Python :: 3",
|
|
]
|
|
dependencies = [
|
|
"av==12.3.0",
|
|
"einops==0.8.0",
|
|
"gdown==5.2.0",
|
|
"gym==0.22.0",
|
|
"hydra-core==1.3.2",
|
|
"imageio==2.35.1",
|
|
"matplotlib==3.7.5",
|
|
"omegaconf==2.3.0",
|
|
"pretty_errors==1.2.25",
|
|
"torch==2.4.0",
|
|
"tqdm==4.66.5",
|
|
"wandb==0.17.7"
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
gym = [
|
|
"cython<3",
|
|
"d4rl",
|
|
"patchelf",
|
|
]
|
|
kitchen = [
|
|
"cython<3",
|
|
"d4rl",
|
|
"dm_control==1.0.16",
|
|
"mujoco==3.1.6",
|
|
"patchelf",
|
|
]
|
|
robomimic = [
|
|
"cython<3",
|
|
"d4rl",
|
|
"patchelf",
|
|
"robomimic @ git+https://github.com/ARISE-Initiative/robomimic.git",
|
|
"robosuite @ git+https://github.com/ARISE-Initiative/robosuite.git@v1.4.1",
|
|
]
|
|
d3il = [
|
|
"cython<3",
|
|
"d4rl",
|
|
"gin-config",
|
|
"patchelf",
|
|
"pin",
|
|
]
|
|
furniture = [
|
|
"cython<3",
|
|
"d4rl",
|
|
"ipdb==0.13.13",
|
|
"patchelf",
|
|
]
|
|
exact = [ # exact likelihood
|
|
"torchdiffeq",
|
|
]
|
|
all = [
|
|
"cython<3",
|
|
"d4rl",
|
|
"gin-config",
|
|
"ipdb==0.13.13",
|
|
"patchelf",
|
|
"pin",
|
|
"robomimic @ git+https://github.com/ARISE-Initiative/robomimic.git",
|
|
"robosuite @ git+https://github.com/ARISE-Initiative/robosuite.git@v1.4.1",
|
|
]
|
|
|
|
[project.urls]
|
|
Homepage = "https://diffusion-ppo.github.io/"
|
|
|
|
[tool.setuptools.packages.find]
|
|
exclude = []
|
|
|
|
[tool.wheel]
|
|
exclude = []
|